Fix issue with chroma when not quantizing

2026-01-26 16:39:47 +00:00 · 2025-10-19 12:13:05 -06:00
parent d51297bcf9
commit b7f85928f3
1 changed files with 1 additions and 0 deletions
--- 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):