Spaces:
Running
on
Zero
Running
on
Zero
Update app.py
Browse files
app.py
CHANGED
@@ -27,7 +27,7 @@ login(token=HF_TOKEN)
|
|
27 |
import diffusers
|
28 |
|
29 |
# init
|
30 |
-
dtype = torch.
|
31 |
device = "cuda" if torch.cuda.is_available() else "cpu"
|
32 |
base_model = "black-forest-labs/FLUX.1-dev"
|
33 |
|
|
|
27 |
import diffusers
|
28 |
|
29 |
# init
|
30 |
+
dtype = torch.bfloat16
|
31 |
device = "cuda" if torch.cuda.is_available() else "cpu"
|
32 |
base_model = "black-forest-labs/FLUX.1-dev"
|
33 |
|