Skip to content

Commit 1fbaff4

Browse files
committed
Fix: Update time_embedder_dtype assignment to use the first parameter's dtype in SkyReelsV2TimeTextImageEmbedding
1 parent 5f2de92 commit 1fbaff4

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

src/diffusers/models/transformers/transformer_skyreels_v2.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -198,7 +198,7 @@ def forward(
198198
):
199199
timestep = self.timesteps_proj(timestep)
200200

201-
time_embedder_dtype = get_parameter_dtype(self.time_embedder)
201+
time_embedder_dtype = next(iter(self.time_embedder.parameters())).dtype
202202
if timestep.dtype != time_embedder_dtype and time_embedder_dtype != torch.int8:
203203
timestep = timestep.to(time_embedder_dtype)
204204
temb = self.time_embedder(timestep).type_as(encoder_hidden_states)

0 commit comments

Comments
 (0)