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