mirror of
https://github.com/hiyouga/LLaMA-Factory.git
synced 2026-08-17 13:35:44 +08:00
[v1] replace custom template system with apply_chat_template (#10598)
Co-authored-by: frozenleaves <frozenleaves@users.noreply.github.com> Co-authored-by: Claude Opus 4.8 <noreply@anthropic.com> Co-authored-by: gemini-code-assist[bot] <176961590+gemini-code-assist[bot]@users.noreply.github.com>
This commit is contained in:
@@ -21,7 +21,7 @@ from ..config import InputArgument, ModelArguments, SampleArguments, SampleBacke
|
||||
from ..core.base_sampler import BaseSampler
|
||||
from ..core.data_engine import DataEngine
|
||||
from ..core.model_engine import ModelEngine
|
||||
from ..core.utils.rendering import Renderer
|
||||
from ..core.rendering import Renderer
|
||||
from ..utils.types import HFModel, Message, Sample, TorchDataset
|
||||
|
||||
|
||||
@@ -118,7 +118,7 @@ def run_chat(args: InputArgument = None):
|
||||
response += new_text
|
||||
|
||||
print()
|
||||
messages.append(model_engine.renderer.parse_message(response))
|
||||
messages.append({"role": "assistant", "content": [{"type": "text", "value": response}]})
|
||||
|
||||
|
||||
if __name__ == "__main__":
|
||||
|
||||
Reference in New Issue
Block a user