mirror of
https://github.com/Wan-Video/Wan2.1.git
synced 2025-06-05 14:54:54 +00:00
update text2video.py
This commit is contained in:
parent
1c7b73d13e
commit
36d6d91b90
@ -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(
|
||||
|
Loading…
Reference in New Issue
Block a user