convert ae to bfloat16

app.py

@@ -29,6 +29,7 @@ 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="cpu" if offload else device)
+    ae.to(dtype=torch.bfloat16)
     return model, ae, t5, clip