Skip to content

GitLab

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
Hide whitespace changes
Inline Side-by-side
Showing with 6 additions and 6 deletions
inference_scripts/inference_pmv2_contronet.py
View file @ 77b17c4f
...@@ -61,9 +61,9 @@ pipe = PhotoMakerStableDiffusionXLControlNetPipeline.from_pretrained( ...@@ -61,9 +61,9 @@ pipe = PhotoMakerStableDiffusionXLControlNetPipeline.from_pretrained(
### Load PhotoMaker checkpoint ### Load PhotoMaker checkpoint
pipe.load_photomaker_adapter( pipe.load_photomaker_adapter(
os.path.dirname(photomaker_path), os.path.dirname(photomaker_ckpt),
subfolder="", subfolder="",
weight_name=os.path.basename(photomaker_path), weight_name=os.path.basename(photomaker_ckpt),
trigger_word="img" # define the trigger word trigger_word="img" # define the trigger word
) )
### Also can cooperate with other LoRA modules ### Also can cooperate with other LoRA modules
......
inference_scripts/inference_pmv2_ip_adapter.py
View file @ 77b17c4f
...@@ -49,9 +49,9 @@ style_images = [load_image(f"./examples/statue.png")] ...@@ -49,9 +49,9 @@ style_images = [load_image(f"./examples/statue.png")]
### Load PhotoMaker checkpoint ### Load PhotoMaker checkpoint
pipe.load_photomaker_adapter( pipe.load_photomaker_adapter(
os.path.dirname(photomaker_path), os.path.dirname(photomaker_ckpt),
subfolder="", subfolder="",
weight_name=os.path.basename(photomaker_path), weight_name=os.path.basename(photomaker_ckpt),
trigger_word="img" # define the trigger word trigger_word="img" # define the trigger word
) )
### Also can cooperate with other LoRA modules ### Also can cooperate with other LoRA modules
......
inference_scripts/inference_pmv2_t2i_adapter.py
View file @ 77b17c4f
...@@ -62,9 +62,9 @@ pipe = PhotoMakerStableDiffusionXLAdapterPipeline.from_pretrained( ...@@ -62,9 +62,9 @@ pipe = PhotoMakerStableDiffusionXLAdapterPipeline.from_pretrained(
### Load PhotoMaker checkpoint ### Load PhotoMaker checkpoint
pipe.load_photomaker_adapter( pipe.load_photomaker_adapter(
os.path.dirname(photomaker_path), os.path.dirname(photomaker_ckpt),
subfolder="", subfolder="",
weight_name=os.path.basename(photomaker_path), weight_name=os.path.basename(photomaker_ckpt),
trigger_word="img" # define the trigger word trigger_word="img" # define the trigger word
) )
### Also can cooperate with other LoRA modules ### 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