Former-commit-id: 8857e45602
This commit is contained in:
hiyouga
2023-09-14 17:56:58 +08:00
parent fe2ac71d11
commit 632fff02e0
4 changed files with 4 additions and 3 deletions

View File

@@ -13,6 +13,7 @@ class ChatModel:
def __init__(self, args: Optional[Dict[str, Any]] = None) -> None:
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.tokenizer.padding_side = "left"
self.model = dispatch_model(self.model)
self.template = get_template_and_fix_tokenizer(data_args.template, self.tokenizer)
self.system_prompt = data_args.system_prompt