Skip to content

GitLab

"docs/zh_cn/vscode:/vscode.git/clone" did not exist on "5840c7655cf442de94f91993fdb0f4f13466c8df"
Unverified Commit 8f1a95d8 authored by 刘悦's avatar 刘悦 Committed by GitHub
Browse files

Fix model v2 link (#156) 

fix model v2 link,default is v1 link
parent 77b17c4f
Hide whitespace changes
Inline Side-by-side
Showing with 1 addition and 1 deletion
gradio_demo/app_v2.py
View file @ 8f1a95d8
...@@ -40,7 +40,7 @@ ASPECT_RATIO_LABELS = list(aspect_ratios) ...@@ -40,7 +40,7 @@ ASPECT_RATIO_LABELS = list(aspect_ratios)
DEFAULT_ASPECT_RATIO = ASPECT_RATIO_LABELS[0] DEFAULT_ASPECT_RATIO = ASPECT_RATIO_LABELS[0]
enable_doodle_arg = False enable_doodle_arg = False
photomaker_ckpt = hf_hub_download(repo_id="TencentARC/PhotoMaker", filename="photomaker-v2.bin", repo_type="model") photomaker_ckpt = hf_hub_download(repo_id="TencentARC/PhotoMaker-V2", filename="photomaker-v2.bin", repo_type="model")
torch_dtype = torch.bfloat16 if torch.cuda.is_bf16_supported() else torch.float16 torch_dtype = torch.bfloat16 if torch.cuda.is_bf16_supported() else torch.float16
if device == "mps": if device == "mps":
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment