MisterAI commited on
Commit
7cc1e07
·
verified ·
1 Parent(s): 13eeca7

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -1
app.py CHANGED
@@ -225,8 +225,9 @@ class FluxGenerator:
225
  "black-forest-labs/FLUX.1-schnell",
226
  revision="refs/pr/1",
227
  torch_dtype=torch.float32 # Utilise float32 au lieu de bfloat16 pour meilleure compatibilité CPU
 
228
  # device_map={"cpu": self.device} # Force tous les composants sur CPU
229
- )device
230
 
231
  # Désactive les optimisations GPU
232
  self.pipeline.to(self.device)
 
225
  "black-forest-labs/FLUX.1-schnell",
226
  revision="refs/pr/1",
227
  torch_dtype=torch.float32 # Utilise float32 au lieu de bfloat16 pour meilleure compatibilité CPU
228
+ )
229
  # device_map={"cpu": self.device} # Force tous les composants sur CPU
230
+ # )device
231
 
232
  # Désactive les optimisations GPU
233
  self.pipeline.to(self.device)