Fix model v2 link (#156)

fix model v2 link,default is v1 link

Fix model v2 link (#156)
fix model v2 link,default is v1 link
8f1a95d8 · 刘悦 · GitHub · 77b17c4f · 8f1a95d8
Unverified Commit 8f1a95d8 authored Jul 24, 2024 by 刘悦 Committed by GitHub Jul 24, 2024
Hide whitespace changes
Inline Side-by-side

Showing with 1 addition and 1 deletion

gradio_demo/app_v2.py gradio_demo/app_v2.py +1 -1

No files found.
--- a/gradio_demo/app_v2.py
+++ b/gradio_demo/app_v2.py
@@ -40,7 +40,7 @@ ASPECT_RATIO_LABELS = list(aspect_ratios)
 DEFAULT_ASPECT_RATIO = ASPECT_RATIO_LABELS[0]

 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
 if device == "mps":