"tests/pipelines/vscode:/vscode.git/clone" did not exist on "98388670d293a590e69d9ff6b442169829f47cf3"
Unverified Commit a08b7284 authored by Zhen Li's avatar Zhen Li Committed by GitHub
Browse files

Update inference_pmv2_ip_adapter.py

parent 803a112d
......@@ -30,7 +30,7 @@ if device == "mps":
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")
prompt = "portrait photo of a woman img, colorful, perfect face, best quality"
negative_prompt = "(asymmetry, worst quality, low quality, illustration), open mouth"
......@@ -98,4 +98,4 @@ images = pipe(
).images
for idx, img in enumerate(images):
img.save(os.path.join(output_dir, f"output_pmv2_ipa_{idx}.jpg"))
\ No newline at end of file
img.save(os.path.join(output_dir, f"output_pmv2_ipa_{idx}.jpg"))
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