From b7f85928f3787e02f9bcd4bb56df1cb0298022ae Mon Sep 17 00:00:00 2001 From: Jaret Burkett Date: Sun, 19 Oct 2025 12:13:05 -0600 Subject: [PATCH] Fix issue with chroma when not quantizing --- extensions_built_in/diffusion_models/chroma/src/layers.py | 1 + 1 file changed, 1 insertion(+) diff --git a/extensions_built_in/diffusion_models/chroma/src/layers.py b/extensions_built_in/diffusion_models/chroma/src/layers.py index 030a57cd..6a668323 100644 --- a/extensions_built_in/diffusion_models/chroma/src/layers.py +++ b/extensions_built_in/diffusion_models/chroma/src/layers.py @@ -403,6 +403,7 @@ class Approximator(nn.Module): return next(self.parameters()).device def forward(self, x: Tensor) -> Tensor: + x = x.to(self.in_proj.weight.dtype) x = self.in_proj(x) for layer, norms in zip(self.layers, self.norms):