Spaces:
Running
on
Zero
Running
on
Zero
zhiweili
commited on
Commit
•
ecf8ba9
1
Parent(s):
c22253e
change to app_haircolor_inpaint_adapter_15
Browse files- app.py +1 -1
- app_haircolor_inpaint_adapter_15.py +3 -2
app.py
CHANGED
@@ -1,6 +1,6 @@
|
|
1 |
import gradio as gr
|
2 |
|
3 |
-
from
|
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 |
-
|
|
|
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 |
|