zhiweili commited on
Commit
1ff5892
1 Parent(s): 4d72328

change base model

Browse files
Files changed (1) hide show
  1. app_haircolor.py +1 -1
app_haircolor.py CHANGED
@@ -19,7 +19,7 @@ from controlnet_aux import (
19
  CannyDetector,
20
  )
21
 
22
- BASE_MODEL = "stabilityai/stable-diffusion-xl-base-1.0"
23
  DEVICE = "cuda" if torch.cuda.is_available() else "cpu"
24
 
25
  DEFAULT_EDIT_PROMPT = "a woman, blue hair, high detailed"
 
19
  CannyDetector,
20
  )
21
 
22
+ BASE_MODEL = "SG161222/RealVisXL_V5.0_Lightning"
23
  DEVICE = "cuda" if torch.cuda.is_available() else "cpu"
24
 
25
  DEFAULT_EDIT_PROMPT = "a woman, blue hair, high detailed"