Fix AE

app.py

@@ -26,7 +26,6 @@ def get_models(name: str, device: torch.device, offload: bool, fp8: bool):
     model = load_flow_model(name, device="cpu" if offload else device)
     model.eval()
     ae = load_ae(name, device=device)
-    ae.to(device=device, dtype=torch.bfloat16)
     return model, ae, t5, clip
 
 
@@ -66,7 +65,7 @@ class FluxGenerator:
         if self.offload:
             self.ae.encoder.to(self.device)
 
-        x = x.to(self.device, dtype=torch.bfloat16)
+        x = x.to(self.device)
 
         # 2) Encode with autocast
         with torch.autocast(device_type=self.device.type, dtype=torch.bfloat16):