Spaces:
Running
on
Zero
Running
on
Zero
Update PhotoMaker v2 demo
Browse files- pipeline_t2i_adapter.py +2 -1
pipeline_t2i_adapter.py
CHANGED
|
@@ -1,5 +1,5 @@
|
|
| 1 |
#####
|
| 2 |
-
# Modified from https://github.com/huggingface/diffusers/blob/v0.29.1/src/diffusers/pipelines/
|
| 3 |
# PhotoMaker v2 @ TencentARC and MCG-NKU
|
| 4 |
# Author: Zhen Li
|
| 5 |
#####
|
|
@@ -249,6 +249,7 @@ class PhotoMakerStableDiffusionXLAdapterPipeline(StableDiffusionXLAdapterPipelin
|
|
| 249 |
if keys != ["id_encoder", "lora_weights"]:
|
| 250 |
raise ValueError("Required keys are (`id_encoder` and `lora_weights`) missing from the state dict.")
|
| 251 |
|
|
|
|
| 252 |
self.trigger_word = trigger_word
|
| 253 |
# load finetuned CLIP image encoder and fuse module here if it has not been registered to the pipeline yet
|
| 254 |
print(f"Loading PhotoMaker {pm_version} components [1] id_encoder from [{pretrained_model_name_or_path_or_dict}]...")
|
|
|
|
| 1 |
#####
|
| 2 |
+
# Modified from https://github.com/huggingface/diffusers/blob/v0.29.1/src/diffusers/pipelines/t2i_adapter/pipeline_stable_diffusion_xl_adapter.py
|
| 3 |
# PhotoMaker v2 @ TencentARC and MCG-NKU
|
| 4 |
# Author: Zhen Li
|
| 5 |
#####
|
|
|
|
| 249 |
if keys != ["id_encoder", "lora_weights"]:
|
| 250 |
raise ValueError("Required keys are (`id_encoder` and `lora_weights`) missing from the state dict.")
|
| 251 |
|
| 252 |
+
self.num_tokens = 2
|
| 253 |
self.trigger_word = trigger_word
|
| 254 |
# load finetuned CLIP image encoder and fuse module here if it has not been registered to the pipeline yet
|
| 255 |
print(f"Loading PhotoMaker {pm_version} components [1] id_encoder from [{pretrained_model_name_or_path_or_dict}]...")
|