mps: fix torch.where in svd codepath.

2025-12-09 22:14:43 +08:00 · 2023-12-18 18:58:09 +01:00
1 changed files with 1 additions and 1 deletions
--- a/src/diffusers/models/resnet.py
+++ b/src/diffusers/models/resnet.py
@@ -1334,7 +1334,7 @@ class AlphaBlender(nn.Module):

            alpha = torch.where(
                image_only_indicator.bool(),
-                torch.ones(1, 1, device=image_only_indicator.device),
+                torch.ones(1, 1, device=image_only_indicator.device, dtype=self.mix_factor.dtype),
                torch.sigmoid(self.mix_factor)[..., None],
            )