fix: diable thinking

ChenZiHong-Gavin · ChenZiHong-Gavin · commit 04a9744d6564 · 2026-02-09T11:36:45.000+08:00
diff --git a/graphgen/models/llm/local/vllm_wrapper.py b/graphgen/models/llm/local/vllm_wrapper.py
@@ -102,6 +102,7 @@ async def generate_topk_per_token(
             temperature=0,
             max_tokens=1,
             logprobs=self.top_k,
+            chat_template_kwargs={"enable_thinking": False}
         )
 
         try:

Original file line number	Diff line number	Diff line change
`@@ -102,6 +102,7 @@ async def generate_topk_per_token(`
`102`	`102`	`temperature=0,`
`103`	`103`	`max_tokens=1,`
`104`	`104`	`logprobs=self.top_k,`
	`105`	`+ chat_template_kwargs={"enable_thinking": False}`
`105`	`106`	`)`
`106`	`107`
`107`	`108`	`try:`