Spaces:
Runtime error
Runtime error
Update app.py
Browse files
app.py
CHANGED
|
@@ -5,8 +5,9 @@ import torch
|
|
| 5 |
import uuid
|
| 6 |
|
| 7 |
# Set device
|
| 8 |
-
device = "cuda" if torch.cuda.is_available() else "cpu"
|
| 9 |
-
dtype = torch.float16 if device == "cuda" else torch.float32
|
|
|
|
| 10 |
|
| 11 |
# Load model
|
| 12 |
pipe = StableDiffusionImg2ImgPipeline.from_pretrained(
|
|
|
|
| 5 |
import uuid
|
| 6 |
|
| 7 |
# Set device
|
| 8 |
+
# device = "cuda" if torch.cuda.is_available() else "cpu"
|
| 9 |
+
# dtype = torch.float16 if device == "cuda" else torch.float32
|
| 10 |
+
dtype = torch.float32
|
| 11 |
|
| 12 |
# Load model
|
| 13 |
pipe = StableDiffusionImg2ImgPipeline.from_pretrained(
|