From 82ee75a969ff315bde680949fb7be7431a965b87 Mon Sep 17 00:00:00 2001 From: Aryan Date: Mon, 23 Dec 2024 08:47:36 +0100 Subject: [PATCH] fix --- src/diffusers/models/embeddings.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/src/diffusers/models/embeddings.py b/src/diffusers/models/embeddings.py index 4558d48edad9..1768c81ce039 100644 --- a/src/diffusers/models/embeddings.py +++ b/src/diffusers/models/embeddings.py @@ -748,10 +748,10 @@ def forward(self, text_embeds: torch.Tensor, image_embeds: torch.Tensor): pos_embedding = self._get_positional_embeddings( height, width, pre_time_compression_frames, device=embeds.device ) - pos_embedding = pos_embedding.to(dtype=embeds.dtype) else: pos_embedding = self.pos_embedding + pos_embedding = pos_embedding.to(dtype=embeds.dtype) embeds = embeds + pos_embedding return embeds