Spaces:
Runtime error
Runtime error
Update app.py
Browse files
app.py
CHANGED
@@ -55,8 +55,8 @@ def download_image(url):
|
|
55 |
|
56 |
#device = "cpu" #"cuda" if torch.cuda.is_available() else "cpu"
|
57 |
|
58 |
-
device = torch.device(
|
59 |
-
print(
|
60 |
# Convert model parameters and buffers to CPU or Cuda
|
61 |
|
62 |
model_id_or_path = "CompVis/stable-diffusion-v1-4"
|
|
|
55 |
|
56 |
#device = "cpu" #"cuda" if torch.cuda.is_available() else "cpu"
|
57 |
|
58 |
+
device = torch.device("cuda:0" if torch.cuda.is_available() else "cpu")
|
59 |
+
print("The model will be running on :: ", device, "device")
|
60 |
# Convert model parameters and buffers to CPU or Cuda
|
61 |
|
62 |
model_id_or_path = "CompVis/stable-diffusion-v1-4"
|