Spaces:

tight-inversion
/

tight-inversion-pulid-demo

Running on Zero

tight-inversion commited on Mar 6

Commit

9499d13

1 Parent(s): a11990f

Test

Files changed (1) hide show

app.py CHANGED Viewed

@@ -135,6 +135,7 @@ class FluxGenerator:
         x = rearrange(x, "h w c -> 1 c h w")  # shape: (1, C, H, W)
         x = x.to(self.device)
         # Encode with autocast
         with torch.autocast(device_type=self.device.type, dtype=torch.bfloat16):
             x = self.ae.encode(x)

         x = rearrange(x, "h w c -> 1 c h w")  # shape: (1, C, H, W)
         x = x.to(self.device)
         # Encode with autocast
+        self.ae.encoder.to(self.device)
         with torch.autocast(device_type=self.device.type, dtype=torch.bfloat16):
             x = self.ae.encode(x)