From bf480574a2d93758185075015fe8d028a01bcf0c Mon Sep 17 00:00:00 2001 From: thinhlpg Date: Fri, 4 Apr 2025 00:54:40 +0700 Subject: [PATCH] fix: minor bug --- train_grpo.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/train_grpo.py b/train_grpo.py index 44b750b..7c7ada7 100644 --- a/train_grpo.py +++ b/train_grpo.py @@ -88,7 +88,7 @@ def agentic_generate( ): # Create agent with appropriate adapter based on tokenizer tokenizer_name = tokenizer.name_or_path.lower() - if "deepseek-r1-distill" in tokenizer_name: + if "deepseek-ai/deepseek-r1-distill" in tokenizer_name: adapter = R1DistilTokenizerAdapter() elif "llama" in tokenizer_name: adapter = LlamaTokenizerAdapter()