Skip to content

Commit d388440

Browse files
committed
fix to prevent load_image_encoder from running for wan 2.2 iv
1 parent 8ef6a0d commit d388440

1 file changed

Lines changed: 1 addition & 1 deletion

File tree

src/maxdiffusion/pipelines/wan/wan_pipeline.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -581,7 +581,7 @@ def _create_common_components(cls, config, vae_only=False, i2v=False):
581581
components["tokenizer"] = cls.load_tokenizer(config=config)
582582
components["text_encoder"] = cls.load_text_encoder(config=config)
583583
components["scheduler"], components["scheduler_state"] = cls.load_scheduler(config=config)
584-
if i2v:
584+
if i2v and config.model_name == 'wan2.1':
585585
components["image_processor"], components["image_encoder"] = cls.load_image_encoder(config)
586586
return components
587587

0 commit comments

Comments
 (0)