Skip to content

Commit a180f06

Browse files
authored
Update diffusers_model_pipeline.py
Fix the bugs after diffusers 0.24 update. The init function should specify the para name. see: huggingface/diffusers#6094
1 parent 032a4f5 commit a180f06

File tree

1 file changed

+9
-9
lines changed

1 file changed

+9
-9
lines changed

src/diffusers_model_pipeline.py

Lines changed: 9 additions & 9 deletions
Original file line numberDiff line numberDiff line change
@@ -557,15 +557,15 @@ def __init__(
557557
modifier_token_id: list = [],
558558
modifier_token_id_2: list = []
559559
):
560-
super().__init__(vae,
561-
text_encoder,
562-
text_encoder_2,
563-
tokenizer,
564-
tokenizer_2,
565-
unet,
566-
scheduler,
567-
force_zeros_for_empty_prompt,
568-
add_watermarker,
560+
super().__init__(vae=vae,
561+
text_encoder=text_encoder,
562+
text_encoder_2=text_encoder_2,
563+
tokenizer=tokenizer,
564+
tokenizer_2=tokenizer_2,
565+
unet=unet,
566+
scheduler=scheduler,
567+
force_zeros_for_empty_prompt=force_zeros_for_empty_prompt,
568+
add_watermarker=add_watermarker,
569569
)
570570

571571
# change attn class

0 commit comments

Comments
 (0)