Upload to Harry Potter

YerongLi · Jul 10, 2023 · e045b1b · e045b1b
1 parent 0c4e2a5
commit e045b1b
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 1 deletion.
diff --git a/src/cli_demo.py b/src/cli_demo.py
@@ -97,6 +97,7 @@ def main():
             continue
         history = truncate_history(history)
         count = 0
+        print(type(model))
         for _, history in model.stream_chat(tokenizer, query, history=history, **generating_args.to_dict()):
             if stop_stream:
                 stop_stream = False

diff --git a/src/upload.py b/src/upload.py
@@ -6,4 +6,4 @@
 from utils import prepare_infer_args, auto_configure_device_map, load_pretrained
 model_args, finetuning_args, generating_args = prepare_infer_args()
 model, tokenizer = load_pretrained(model_args, finetuning_args)
-model.push_to_hub("chatglmv1_hp")
+model.push_to_hub("harry_potter")