update text2video.py

This commit is contained in:
Yexiong Lin 2025-03-04 19:54:54 +11:00
parent 1c7b73d13e
commit 36d6d91b90

View File

@ -298,8 +298,6 @@ class WanT2V:
timestep = [t]
timestep = torch.stack(timestep)
self.model.to(self.device)
noise_pred_cond = self.model(
latent_model_input, t=timestep, **arg_c)[0]
noise_pred_uncond = self.model(