Update web_demo.py

2023-07-14 00:54:46 +08:00 · 2023-07-14 00:54:46 +08:00 · 0676254266
parent 830c0cc2a8
commit 0676254266
1 changed files with 1 additions and 0 deletions
--- a/web_demo.py
+++ b/web_demo.py
@ -205,6 +205,7 @@ def parse_args():

 def main(args):
    torch.cuda.empty_cache()
+    
    print("load model begin.")
    model, tokenizer = load_model(args.model_path)
    print("load model end.")