zhiweili commited on
Commit
ecf8ba9
1 Parent(s): c22253e

change to app_haircolor_inpaint_adapter_15

Browse files
Files changed (2) hide show
  1. app.py +1 -1
  2. app_haircolor_inpaint_adapter_15.py +3 -2
app.py CHANGED
@@ -1,6 +1,6 @@
1
  import gradio as gr
2
 
3
- from app_masked import create_demo as create_demo_haircolor
4
 
5
  with gr.Blocks(css="style.css") as demo:
6
  with gr.Tabs():
 
1
  import gradio as gr
2
 
3
+ from app_haircolor_inpaint_adapter_15 import create_demo as create_demo_haircolor
4
 
5
  with gr.Blocks(css="style.css") as demo:
6
  with gr.Tabs():
app_haircolor_inpaint_adapter_15.py CHANGED
@@ -26,8 +26,9 @@ from controlnet_aux import (
26
  HEDdetector,
27
  )
28
 
29
- BASE_MODEL = "stable-diffusion-v1-5/stable-diffusion-v1-5"
30
- # BASE_MODEL = "stable-diffusion-v1-5/stable-diffusion-inpainting"
 
31
 
32
  DEVICE = "cuda" if torch.cuda.is_available() else "cpu"
33
 
 
26
  HEDdetector,
27
  )
28
 
29
+ # BASE_MODEL = "stable-diffusion-v1-5/stable-diffusion-v1-5"
30
+ BASE_MODEL = "stable-diffusion-v1-5/stable-diffusion-inpainting"
31
+ # BASE_MODEL = "refiners/sd15.juggernaut.aftermath.unet_inpainting"
32
 
33
  DEVICE = "cuda" if torch.cuda.is_available() else "cpu"
34