GenDoc_05

Running

MisterAI commited on Feb 15

Commit

e372698

verified ·

1 Parent(s): 9d225f0

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -81,7 +81,7 @@ class FluxGenerator:
                 "black-forest-labs/FLUX.1-schnell",
                 revision="refs/pr/1",
                 device_map="balanced",
-                torch_dtype=torch.float32,
                 use_safetensors=True
 )

                 "black-forest-labs/FLUX.1-schnell",
                 revision="refs/pr/1",
                 device_map="balanced",
+                torch_dtype=torch.float8,
                 use_safetensors=True
 )