Unverified Commit 77b17c4f authored by Clay Chai's avatar Clay Chai Committed by GitHub
Browse files

[fix] checkpoint path (#154)

* fix checkpoint path

* fix checkpoint path in inference_pmv2_controlnet
parent 0b13e0a3
......@@ -61,9 +61,9 @@ pipe = PhotoMakerStableDiffusionXLControlNetPipeline.from_pretrained(
### Load PhotoMaker checkpoint
pipe.load_photomaker_adapter(
os.path.dirname(photomaker_path),
os.path.dirname(photomaker_ckpt),
subfolder="",
weight_name=os.path.basename(photomaker_path),
weight_name=os.path.basename(photomaker_ckpt),
trigger_word="img" # define the trigger word
)
### Also can cooperate with other LoRA modules
......
......@@ -49,9 +49,9 @@ style_images = [load_image(f"./examples/statue.png")]
### Load PhotoMaker checkpoint
pipe.load_photomaker_adapter(
os.path.dirname(photomaker_path),
os.path.dirname(photomaker_ckpt),
subfolder="",
weight_name=os.path.basename(photomaker_path),
weight_name=os.path.basename(photomaker_ckpt),
trigger_word="img" # define the trigger word
)
### Also can cooperate with other LoRA modules
......
......@@ -62,9 +62,9 @@ pipe = PhotoMakerStableDiffusionXLAdapterPipeline.from_pretrained(
### Load PhotoMaker checkpoint
pipe.load_photomaker_adapter(
os.path.dirname(photomaker_path),
os.path.dirname(photomaker_ckpt),
subfolder="",
weight_name=os.path.basename(photomaker_path),
weight_name=os.path.basename(photomaker_ckpt),
trigger_word="img" # define the trigger word
)
### Also can cooperate with other LoRA modules
......
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