Spaces:
Running
on
Zero
Running
on
Zero
Anurag Bhardwaj
commited on
Update app.py
Browse files
app.py
CHANGED
@@ -17,7 +17,7 @@ def load_pipeline():
|
|
17 |
base_model = "black-forest-labs/FLUX.1-dev"
|
18 |
pipe = DiffusionPipeline.from_pretrained(
|
19 |
base_model,
|
20 |
-
torch_dtype=torch.
|
21 |
)
|
22 |
|
23 |
# Load LoRA weights
|
|
|
17 |
base_model = "black-forest-labs/FLUX.1-dev"
|
18 |
pipe = DiffusionPipeline.from_pretrained(
|
19 |
base_model,
|
20 |
+
torch_dtype=torch.float32
|
21 |
)
|
22 |
|
23 |
# Load LoRA weights
|