Hugging Face's logo

Spaces:
smartfeed
/
image2image
Runtime error

App Files Community
zhiweili commited on
Commit
7a7ec30
·
1 Parent(s): e2d5eb3

change base model

Browse files
Files changed (1) hide show
  1. 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