"examples/vscode:/vscode.git/clone" did not exist on "5ff25cdf5b1310e83d9e595142b39ae4d7b561e9"
Commit 9196a220 authored by helloyongyang's avatar helloyongyang
Browse files

remove mask in wan22 5B t2v

parent b469676c
...@@ -60,12 +60,9 @@ class WanScheduler(BaseScheduler): ...@@ -60,12 +60,9 @@ class WanScheduler(BaseScheduler):
device=self.device, device=self.device,
generator=self.generator, generator=self.generator,
) )
if self.config["model_cls"] == "wan2.2": if self.config["model_cls"] == "wan2.2" and self.config["task"] == "i2v":
if self.config["task"] == "t2v": self.mask = masks_like(self.latents, zero=True)
self.mask = masks_like(self.latents, zero=False) self.latents = (1.0 - self.mask) * self.vae_encoder_out + self.mask * self.latents
elif self.config["task"] == "i2v":
self.mask = masks_like(self.latents, zero=True)
self.latents = (1.0 - self.mask) * self.vae_encoder_out + self.mask * self.latents
def set_timesteps( def set_timesteps(
self, self,
......
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