From fb04e5f6e0f5bd2e630a0bf08a36a4870cc526d9 Mon Sep 17 00:00:00 2001 From: Jianxin Ma Date: Thu, 3 Aug 2023 21:16:17 +0800 Subject: [PATCH] Add files via upload --- examples/transformers_agent.md | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/examples/transformers_agent.md b/examples/transformers_agent.md index e301ce2..017fcc4 100644 --- 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,