Spaces:
Sleeping
Sleeping
Update app.py
Browse files
app.py
CHANGED
|
@@ -59,7 +59,7 @@ def pretty_features(row: dict) -> dict:
|
|
| 59 |
from diffusers import DiffusionPipeline
|
| 60 |
|
| 61 |
device = "cpu"
|
| 62 |
-
pipe = DiffusionPipeline.from_pretrained(MODEL_ID,
|
| 63 |
pipe.to(device)
|
| 64 |
if hasattr(pipe, "enable_attention_slicing"):
|
| 65 |
pipe.enable_attention_slicing()
|
|
|
|
| 59 |
from diffusers import DiffusionPipeline
|
| 60 |
|
| 61 |
device = "cpu"
|
| 62 |
+
pipe = DiffusionPipeline.from_pretrained(MODEL_ID, torch_dtype=torch.float32)
|
| 63 |
pipe.to(device)
|
| 64 |
if hasattr(pipe, "enable_attention_slicing"):
|
| 65 |
pipe.enable_attention_slicing()
|