Hugging Face's logo

Spaces:
smartfeed
/
image2image
Running on Zero

App Files Community
zhiweili commited on
Commit
7dc06b4
•
1 Parent(s): 7bd62db

change base model

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