Spaces:
Runtime error
Runtime error
treasuraid
commited on
Commit
·
0185ba1
1
Parent(s):
73ba89d
Update app.py
Browse files
app.py
CHANGED
@@ -22,7 +22,7 @@ def load_pipeline_w_lora() :
|
|
22 |
pipeline = DiffusionPipeline.from_pretrained(
|
23 |
MODEL_REPO,
|
24 |
revision=None,
|
25 |
-
torch_dtype=torch.
|
26 |
)
|
27 |
|
28 |
# Load LoRa attn layer weights to unet attn layers
|
|
|
22 |
pipeline = DiffusionPipeline.from_pretrained(
|
23 |
MODEL_REPO,
|
24 |
revision=None,
|
25 |
+
torch_dtype=torch.float32,
|
26 |
)
|
27 |
|
28 |
# Load LoRa attn layer weights to unet attn layers
|