fix: minor bug

main
thinhlpg 1 month ago
parent 3081d6e36b
commit bf480574a2

@ -88,7 +88,7 @@ def agentic_generate(
): ):
# Create agent with appropriate adapter based on tokenizer # Create agent with appropriate adapter based on tokenizer
tokenizer_name = tokenizer.name_or_path.lower() 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() adapter = R1DistilTokenizerAdapter()
elif "llama" in tokenizer_name: elif "llama" in tokenizer_name:
adapter = LlamaTokenizerAdapter() adapter = LlamaTokenizerAdapter()

Loading…
Cancel
Save