Update inference_pmv2_t2i_adapter.py

TencentARC · Jul 22, 2024 · 0b13e0a · 0b13e0a
1 parent a08b728
commit 0b13e0a
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/inference_scripts/inference_pmv2_t2i_adapter.py b/inference_scripts/inference_pmv2_t2i_adapter.py
@@ -32,7 +32,7 @@
 output_dir = "./outputs"
 os.makedirs(output_dir, exist_ok=True)
 
-photomaker_path = 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")
 
 openpose = OpenposeDetector.from_pretrained("lllyasviel/ControlNet")
 # load adapter
@@ -112,4 +112,4 @@
 ).images
 
 for idx, img in enumerate(images): 
-    img.save(os.path.join(output_dir, f"output_pmv2_t2ia_{idx}.jpg"))
+    img.save(os.path.join(output_dir, f"output_pmv2_t2ia_{idx}.jpg"))