github/ai-toolkit
1
0
Fork 0
Code Issues Pull Requests Actions 1 Packages Projects Releases Wiki Activity

Fix issue with chroma when not quantizing

Browse Source
This commit is contained in:
Jaret Burkett
2025-10-19 12:13:05 -06:00
parent d51297bcf9
commit b7f85928f3
1 changed files with 1 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

1
extensions_built_in/diffusion_models/chroma/src/layers.py
View File

@@ -403,6 +403,7 @@ class Approximator(nn.Module):
return next(self.parameters()).device return next(self.parameters()).device
def forward(self, x: Tensor) -> Tensor: def forward(self, x: Tensor) -> Tensor:
x = x.to(self.in_proj.weight.dtype)
x = self.in_proj(x) x = self.in_proj(x)
for layer, norms in zip(self.layers, self.norms): for layer, norms in zip(self.layers, self.norms):