yumoqing/llmengine
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

bugfix

Browse Source
This commit is contained in:
yumoqing 2025-07-24 14:31:45 +08:00
parent 0ca1fad8ee
commit d1b26a5646
1 changed files with 1 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1
llmengine/qwen3.py
View File

@ -37,6 +37,7 @@ class Qwen3LLM(BaseChatLLM):
add_generation_prompt=True, add_generation_prompt=True,
enable_thinking=True enable_thinking=True
) )
debug(f'{text=}, {type(text)=}')
return self.tokenizer([text], return_tensors="pt").to(self.model.device) return self.tokenizer([text], return_tensors="pt").to(self.model.device)
llm_register("Qwen/Qwen3", Qwen3LLM) llm_register("Qwen/Qwen3", Qwen3LLM)