Spaces:
Sleeping
Sleeping
Update app.py
Browse files
app.py
CHANGED
|
@@ -106,7 +106,7 @@ model_path = "CompVis/stable-diffusion-v1-4"
|
|
| 106 |
sd_pipeline = DiffusionPipeline.from_pretrained(
|
| 107 |
model_path,
|
| 108 |
low_cpu_mem_usage=True,
|
| 109 |
-
torch_dtype=torch.
|
| 110 |
).to(torch_device)
|
| 111 |
|
| 112 |
# Load textual inversions
|
|
|
|
| 106 |
sd_pipeline = DiffusionPipeline.from_pretrained(
|
| 107 |
model_path,
|
| 108 |
low_cpu_mem_usage=True,
|
| 109 |
+
torch_dtype=torch.float32
|
| 110 |
).to(torch_device)
|
| 111 |
|
| 112 |
# Load textual inversions
|