Spaces:
Running
on
Zero
Running
on
Zero
Commit
•
58b2d1a
1
Parent(s):
c78df80
Update app.py
Browse files
app.py
CHANGED
|
@@ -27,7 +27,7 @@ os.environ["HF_HOME"] = cache_path
|
|
| 27 |
|
| 28 |
torch.backends.cuda.matmul.allow_tf32 = True
|
| 29 |
|
| 30 |
-
dtype = torch.
|
| 31 |
pipe = FluxWithCFGPipeline.from_pretrained(
|
| 32 |
"ostris/OpenFLUX.1", torch_dtype=dtype
|
| 33 |
).to("cuda")
|
|
|
|
| 27 |
|
| 28 |
torch.backends.cuda.matmul.allow_tf32 = True
|
| 29 |
|
| 30 |
+
dtype = torch.bfloat16
|
| 31 |
pipe = FluxWithCFGPipeline.from_pretrained(
|
| 32 |
"ostris/OpenFLUX.1", torch_dtype=dtype
|
| 33 |
).to("cuda")
|