Spaces:
Runtime error
Runtime error
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 |
|