From 0676254266034e44559561b9e1aaab7bd6f245f3 Mon Sep 17 00:00:00 2001 From: LYMDLUT <70597027+LYMDLUT@users.noreply.github.com> Date: Fri, 14 Jul 2023 00:54:46 +0800 Subject: [PATCH] Update web_demo.py --- web_demo.py | 1 + 1 file changed, 1 insertion(+) diff --git a/web_demo.py b/web_demo.py index 8e9f298..8c6e74c 100644 --- 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.")