[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:
浮梦
2026-07-10 21:15:44 +08:00
committed by GitHub
parent ea31c43d80
commit b61140db3e
28 changed files with 895 additions and 974 deletions

View File

@@ -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__":