Skip to content

Commit 3c0c521

Browse files
committed
support Wan2.2-VACE-Fun-A14B
1 parent d162ace commit 3c0c521

File tree

1 file changed

+4
-4
lines changed

1 file changed

+4
-4
lines changed

scripts/convert_wan_to_diffusers.py

Lines changed: 4 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -998,23 +998,23 @@ def get_args():
998998
image_encoder=image_encoder,
999999
image_processor=image_processor,
10001000
)
1001-
elif "VACE" in args.model_type:
1001+
elif "Wan2.2-VACE" in args.model_type:
10021002
pipe = WanVACEPipeline(
10031003
transformer=transformer,
1004+
transformer_2=transformer_2,
10041005
text_encoder=text_encoder,
10051006
tokenizer=tokenizer,
10061007
vae=vae,
10071008
scheduler=scheduler,
1009+
boundary_ratio=0.875,
10081010
)
1009-
elif "Wan2.2-VACE" in args.model_type:
1011+
elif "VACE" in args.model_type:
10101012
pipe = WanVACEPipeline(
10111013
transformer=transformer,
1012-
transformer_2=transformer_2,
10131014
text_encoder=text_encoder,
10141015
tokenizer=tokenizer,
10151016
vae=vae,
10161017
scheduler=scheduler,
1017-
boundary_ratio=0.875,
10181018
)
10191019
else:
10201020
pipe = WanPipeline(

0 commit comments

Comments
 (0)