Spaces:
Running
on
Zero
Running
on
Zero
zhiweili
commited on
Commit
•
e2d5eb3
1
Parent(s):
61f3bdc
change base model
Browse files- app_ddim.py +2 -1
app_ddim.py
CHANGED
|
@@ -16,7 +16,8 @@ from diffusers import (
|
|
| 16 |
DDIMScheduler,
|
| 17 |
)
|
| 18 |
|
| 19 |
-
BASE_MODEL = "stable-diffusion-v1-5/stable-diffusion-v1-5"
|
|
|
|
| 20 |
|
| 21 |
DEVICE = "cuda" if torch.cuda.is_available() else "cpu"
|
| 22 |
|
|
|
|
| 16 |
DDIMScheduler,
|
| 17 |
)
|
| 18 |
|
| 19 |
+
# BASE_MODEL = "stable-diffusion-v1-5/stable-diffusion-v1-5"
|
| 20 |
+
BASE_MODEL = "SG161222/Realistic_Vision_V5.1_noVAE"
|
| 21 |
|
| 22 |
DEVICE = "cuda" if torch.cuda.is_available() else "cpu"
|
| 23 |
|