update UI, fix #212

Former-commit-id: 4d1641c1bf
This commit is contained in:
hiyouga
2023-07-20 22:09:06 +08:00
parent 6552b74005
commit 94f2fd634f
12 changed files with 155 additions and 45 deletions

View File

@@ -20,7 +20,7 @@ class ChatModel:
) -> None:
self.model, self.tokenizer = load_model_and_tokenizer(model_args, finetuning_args)
self.template = get_template(data_args.prompt_template)
self.source_prefix = data_args.source_prefix if data_args.source_prefix else ""
self.source_prefix = data_args.source_prefix or ""
self.generating_args = generating_args
def process_args(