We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
1 parent a30ed90 commit b8d7d55Copy full SHA for b8d7d55
1 file changed
diffsynth/models/wan_video_dit.py
@@ -362,7 +362,7 @@ def forward(self,
362
**kwargs,
363
):
364
t = self.time_embedding(
365
- sinusoidal_embedding_1d(self.freq_dim, timestep))
+ sinusoidal_embedding_1d(self.freq_dim, timestep).to(x.dtype))
366
t_mod = self.time_projection(t).unflatten(1, (6, self.dim))
367
context = self.text_embedding(context)
368
0 commit comments