diff --git a/comfy/ldm/chroma/layers.py b/comfy/ldm/chroma/layers.py index 606d9688c..8ad3c72d4 100644 --- a/comfy/ldm/chroma/layers.py +++ b/comfy/ldm/chroma/layers.py @@ -126,7 +126,7 @@ class Approximator(nn.Module): def device(self): # Get the device of the module (assumes all parameters are on the same device) return next(self.parameters()).device - + def forward(self, x: Tensor) -> Tensor: x = self.in_proj(x)