Skip to content

GitLab

Commit ae112568 authored by gushiqiao's avatar gushiqiao Committed by GitHub
Browse files

Fix bugs 

Fix bugs
parents e1f7729e 1bbe8c6f
Show whitespace changes
Inline Side-by-side
Showing with 3 additions and 3 deletions
lightx2v/models/runners/default_runner.py
View file @ ae112568
...@@ -176,7 +176,7 @@ class DefaultRunner(BaseRunner): ...@@ -176,7 +176,7 @@ class DefaultRunner(BaseRunner):
self.model = self.load_transformer() self.model = self.load_transformer()
self.init_scheduler() self.init_scheduler()
self.model.scheduler.prepare(self.inputs["image_encoder_output"]) self.model.scheduler.prepare(self.inputs["image_encoder_output"])
if self.config.get("model_cls") == "wan2.2": if self.config.get("model_cls") == "wan2.2" and self.config["task"] == "i2v":
self.inputs["image_encoder_output"]["vae_encoder_out"] = None self.inputs["image_encoder_output"]["vae_encoder_out"] = None
latents, generator = self.run() latents, generator = self.run()
self.end_run() self.end_run()
......
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