Spaces:

MisterAI
/

GenDoc_05

Sleeping

MisterAI commited on Feb 15

Commit

1bd86ef

verified ·

1 Parent(s): 65dd249

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -71,7 +71,7 @@ class FluxGenerator:
             self.pipeline = FluxPipeline.from_pretrained(
                 "black-forest-labs/FLUX.1-schnell",
                 revision="refs/pr/1",
-                device_map="cpu",  # Utilise une chaîne de caractères au lieu d'un dictionnaire
                 **model_kwargs
             )

             self.pipeline = FluxPipeline.from_pretrained(
                 "black-forest-labs/FLUX.1-schnell",
                 revision="refs/pr/1",
+                device_map="balanced",  # Utilise une chaîne de caractères au lieu d'un dictionnaire
                 **model_kwargs
             )