diff --git a/wan/text2video.py b/wan/text2video.py index 4460b8b..99716bf 100644 --- a/wan/text2video.py +++ b/wan/text2video.py @@ -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(