minor fix

This commit is contained in:
mi804
2025-07-25 17:24:42 +08:00
parent 9015d08927
commit bba44173d2
2 changed files with 0 additions and 2 deletions

View File

@@ -619,7 +619,6 @@ class WanVideoPipeline(BasePipeline):
for progress_id, timestep in enumerate(progress_bar_cmd(self.scheduler.timesteps)):
# switch high_noise DiT to low_noise DiT
if models.get("dit2") is not None and timestep.item() < boundary * self.scheduler.num_train_timesteps:
print("switching to low noise DiT")
self.load_models_to_device(["dit2", "motion_controller", "vace"])
models["dit"] = models.pop("dit2")
timestep = timestep.unsqueeze(0).to(dtype=self.torch_dtype, device=self.device)