support pass model_revision with 'None' as using default

1 year ago · 4a47bf559d
parent 164aabd0b9
commit 4a47bf559d
1 changed files with 1 additions and 1 deletions
--- a/web_demo.py
+++ b/web_demo.py
@ -28,7 +28,7 @@ model = AutoModelForCausalLM.from_pretrained(
    device_map="auto",
    trust_remote_code=True,
    resume_download=True,
-    **{"revision": args.model_revision} if args.model_revision is not None and args.model_revision != "" else {},
+    **{"revision": args.model_revision} if args.model_revision is not None and args.model_revision != "" and args.model_revision != "None" else {},
 ).eval()

 model.generation_config = GenerationConfig.from_pretrained("Qwen/Qwen-7B-Chat", trust_remote_code=True, resume_download=True)