fix #761
This commit is contained in:
1 parent
8ea32e4046
commit
b34797a845
3 files changed
+5
-3
No files matched your search
@@ -14,7 +14,6 @@ class ChatModel:
|
||||
model_args, data_args, finetuning_args, self.generating_args = get_infer_args(args)
|
||||
self.model, self.tokenizer = load_model_and_tokenizer(model_args, finetuning_args)
|
||||
self.model = dispatch_model(self.model)
|
||||
self.model = self.model.eval() # enable evaluation mode
|
||||
self.template = get_template_and_fix_tokenizer(data_args.template, self.tokenizer)
|
||||
self.system_prompt = data_args.system_prompt
|
||||
|
||||
|
||||
Reference in new issue
Block a user