Spaces:
Runtime error
Runtime error
HardWorkingStation commited on
Commit ·
72a3fae
1
Parent(s): 99748b4
Initial commit
Browse files
app.py
CHANGED
|
@@ -13,7 +13,7 @@ st.set_page_config(layout="wide")
|
|
| 13 |
st.title('Play with Stable-Diffusion v1-4')
|
| 14 |
|
| 15 |
model_id = "CompVis/stable-diffusion-v1-4"
|
| 16 |
-
device = "cuda"
|
| 17 |
auth_token = os.environ.get("StableDiffusion")
|
| 18 |
|
| 19 |
|
|
|
|
| 13 |
st.title('Play with Stable-Diffusion v1-4')
|
| 14 |
|
| 15 |
model_id = "CompVis/stable-diffusion-v1-4"
|
| 16 |
+
device = "cuda" if torch.cuda.is_available() else "cpu"
|
| 17 |
auth_token = os.environ.get("StableDiffusion")
|
| 18 |
|
| 19 |
|