GenDoc_05

Running

MisterAI commited on Feb 15

Commit

711e1a9

verified ·

1 Parent(s): bfdb818

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -49,7 +49,7 @@ class FluxGenerator:
                 "black-forest-labs/FLUX.1-schnell",
                 revision="refs/pr/1",
                 torch_dtype=torch.float32,  # Utilise float32 au lieu de bfloat16 pour meilleure compatibilité CPU
-                device_map={"": self.device}  # Force tous les composants sur CPU
             )
             # Désactive les optimisations GPU

                 "black-forest-labs/FLUX.1-schnell",
                 revision="refs/pr/1",
                 torch_dtype=torch.float32,  # Utilise float32 au lieu de bfloat16 pour meilleure compatibilité CPU
+                device_map={"cpu": self.device}  # Force tous les composants sur CPU
             )
             # Désactive les optimisations GPU