Hugging Face's logo

Spaces:
smartfeed
/
image2image
Running

App Files Community
zhiweili commited on
Commit
d36033b
·
1 Parent(s): 2f57cfb

change base model

Browse files
Files changed (1) hide show
  1. app_haircolor_inpaint_15.py +1 -1
app_haircolor_inpaint_15.py CHANGED
@@ -24,7 +24,7 @@ from controlnet_aux import (
24
  )
25
 
26
  # BASE_MODEL = "stable-diffusion-v1-5/stable-diffusion-v1-5"
27
- BASE_MODEL = "SG161222/Realistic_Vision_V2.0"
28
 
29
  DEVICE = "cuda" if torch.cuda.is_available() else "cpu"
30
 
 
24
  )
25
 
26
  # BASE_MODEL = "stable-diffusion-v1-5/stable-diffusion-v1-5"
27
+ BASE_MODEL = "SG161222/Realistic_Vision_V5.1_noVAE"
28
 
29
  DEVICE = "cuda" if torch.cuda.is_available() else "cpu"
30