This commit is contained in:
yumoqing 2025-09-16 14:36:17 +08:00
parent b2e4c0befb
commit 542e5da159

View File

@ -40,4 +40,4 @@ class Qwen3CoderLLM(BaseChatLLM):
debug(f'{text=}, {type(text)=}')
return self.tokenizer([text], return_tensors="pt").to(self.model.device)
llm_register("Qwen/Qwen3Coder", Qwen3CoderLLM)
llm_register("Qwen3Coder", Qwen3CoderLLM)