423A35C7/LLaMA-Factory
1
0
Fork 0
mirror of https://github.com/hiyouga/LLaMA-Factory.git synced 2025-12-15 03:10:35 +08:00
Code Issues Packages Projects Releases Wiki Activity

update webui

Browse Source 
Former-commit-id: 3a720aac66
This commit is contained in:
hiyouga
2023-08-09 00:26:11 +08:00
parent 77aa9853fb
commit 4f714ba314
5 changed files with 24 additions and 6 deletions
Download Patch File Download Diff File Expand all files Collapse all files

1
src/llmtuner/chat/stream_chat.py
View File

@@ -19,7 +19,6 @@ class ChatModel:
self.template = get_template_and_fix_tokenizer(data_args.template, self.tokenizer)
self.source_prefix = data_args.source_prefix
self.stop_ids = self.tokenizer.convert_tokens_to_ids(self.template.stop_words)
self.tokenizer.add_special_tokens(dict(additional_special_tokens=self.template.stop_words))
self.model.generate = MethodType(PreTrainedModel.generate, self.model) # disable custom method (for Qwen)
def process_args(