Add files via upload

2 years ago · fb04e5f6e0
parent 7d4614dba2
commit fb04e5f6e0
1 changed files with 1 additions and 1 deletions
--- a/examples/transformers_agent.md
+++ b/examples/transformers_agent.md
@ -47,7 +47,7 @@ class QWenAgent(Agent):
        self.tokenizer = AutoTokenizer.from_pretrained(checkpoint, trust_remote_code=True)
        self.model = AutoModelForCausalLM.from_pretrained(checkpoint, device_map="auto", trust_remote_code=True).cuda().eval()
        self.model.generation_config = GenerationConfig.from_pretrained(checkpoint, trust_remote_code=True) # 可指定不同的生成长度、top_p等相关超参
-        model.generation_config.do_sample = False  # greedy
+        self.model.generation_config.do_sample = False  # greedy
        
        super().__init__(
            chat_prompt_template=chat_prompt_template,