We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent 5f2de92 commit 1fbaff4Copy full SHA for 1fbaff4
src/diffusers/models/transformers/transformer_skyreels_v2.py
@@ -198,7 +198,7 @@ def forward(
198
):
199
timestep = self.timesteps_proj(timestep)
200
201
- time_embedder_dtype = get_parameter_dtype(self.time_embedder)
+ time_embedder_dtype = next(iter(self.time_embedder.parameters())).dtype
202
if timestep.dtype != time_embedder_dtype and time_embedder_dtype != torch.int8:
203
timestep = timestep.to(time_embedder_dtype)
204
temb = self.time_embedder(timestep).type_as(encoder_hidden_states)
0 commit comments