423A35C7/LLaMA-Factory
1
0
Fork 0
mirror of https://github.com/hiyouga/LLaMA-Factory.git synced 2025-08-22 22:02:51 +08:00
Code Issues Packages Projects Releases Wiki Activity

Update hf_engine.py

Browse Source 
Former-commit-id: a943a1034b0033e2fae72e3d272817e3adb03fd1
This commit is contained in:
hoshi-hiyouga 2024-05-20 00:30:45 +08:00 committed by GitHub
parent a710d97748
commit 30b2ec7025
1 changed files with 1 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1
src/llamafactory/chat/hf_engine.py
View File

@ -59,6 +59,7 @@ class HuggingfaceEngine(BaseEngine):
messages[0]["content"] = "<image>" + messages[0]["content"]
paired_messages = messages + [{"role": "assistant", "content": ""}]
system = system or generating_args["default_system"]
prompt_ids, _ = template.encode_oneturn(
tokenizer=tokenizer, messages=paired_messages, system=system, tools=tools
)