starlight-apk/Qwen
1
0
Fork 0
mirror of https://github.com/QwenLM/Qwen.git synced 2026-05-20 08:25:47 +08:00
Code Issues Packages Projects Releases Wiki Activity

support pass model_revision with 'None' as using default

Browse Source
This commit is contained in:
wsl-wy
2023-08-08 02:52:43 +08:00
parent 164aabd0b9
commit 4a47bf559d
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
web_demo.py
View File

@@ -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)