Force free diffusion model to float32 on CPU

app.py

@@ -1033,8 +1033,12 @@ def free_diffusion_png(spec: AssetSpec, index: int, run_id: int) -> tuple[bytes
         from diffusers import DiffusionPipeline
 
         if FREE_DIFFUSION_PIPE is None:
-            FREE_DIFFUSION_PIPE = DiffusionPipeline.from_pretrained(FREE_IMAGE_MODEL)
+            FREE_DIFFUSION_PIPE = DiffusionPipeline.from_pretrained(FREE_IMAGE_MODEL, torch_dtype=torch.float32)
             FREE_DIFFUSION_PIPE = FREE_DIFFUSION_PIPE.to("cpu")
+            for component_name in ("unet", "vae", "text_encoder"):
+                component = getattr(FREE_DIFFUSION_PIPE, component_name, None)
+                if component is not None and hasattr(component, "to"):
+                    component.to(device="cpu", dtype=torch.float32)
             if hasattr(FREE_DIFFUSION_PIPE, "enable_attention_slicing"):
                 FREE_DIFFUSION_PIPE.enable_attention_slicing()