hyperzlib
/
Isekai-Qwen
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

support pass model_revision with 'None' as using default

Browse Source
main
wsl-wy 2 years ago
parent 164aabd0b9
commit 4a47bf559d
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File

2
web_demo.py
Unescape Escape View File

@ -28,7 +28,7 @@ model = AutoModelForCausalLM.from_pretrained(
device_map="auto", device_map="auto",
trust_remote_code=True, trust_remote_code=True,
resume_download=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() ).eval()
model.generation_config = GenerationConfig.from_pretrained("Qwen/Qwen-7B-Chat", trust_remote_code=True, resume_download=True) model.generation_config = GenerationConfig.from_pretrained("Qwen/Qwen-7B-Chat", trust_remote_code=True, resume_download=True)

Write Preview
Loading…
Cancel
Save