ros/LlamaFactory
1
0
Fork 0
mirror of https://github.com/hiyouga/LlamaFactory.git synced 2026-02-02 20:43:38 +00:00
Code Issues Packages Projects Releases Wiki Activity

release v0.7.0

Browse Source 
Former-commit-id: 45bb89cb4d26a6b3fb5360bc90ab950738fe4920
This commit is contained in:
hiyouga
2024-04-26 23:18:00 +08:00
parent cb8b8f40cd
commit c501f377dd
8 changed files with 86 additions and 36 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
src/llmtuner/chat/hf_engine.py
View File

@@ -56,7 +56,7 @@ class HuggingfaceEngine(BaseEngine):
input_kwargs: Optional[Dict[str, Any]] = {},
) -> Tuple[Dict[str, Any], int]:
if processor is not None and image is not None and "<image>" not in messages[0]["content"]:
messages[0]["content"] = messages[0]["content"] + "<image>"
messages[0]["content"] = "<image>" + messages[0]["content"]
paired_messages = messages + [{"role": "assistant", "content": ""}]
prompt_ids, _ = template.encode_oneturn(