Maria
commited on
Commit
·
bff87b0
1
Parent(s):
2387a4c
hw6
Browse files
app.py
CHANGED
|
@@ -26,7 +26,7 @@ def on_checkbox_change(use_advanced):
|
|
| 26 |
|
| 27 |
with gr.Blocks(css=css) as demo:
|
| 28 |
with gr.Column(elem_id="col-container"):
|
| 29 |
-
gr.Markdown(" # Maria Lashina
|
| 30 |
|
| 31 |
MODEL_LIST = [
|
| 32 |
"CompVis/stable-diffusion-v1-4",
|
|
|
|
| 26 |
|
| 27 |
with gr.Blocks(css=css) as demo:
|
| 28 |
with gr.Column(elem_id="col-container"):
|
| 29 |
+
gr.Markdown(" # Maria Lashina T2I Rat Stickers Generation App")
|
| 30 |
|
| 31 |
MODEL_LIST = [
|
| 32 |
"CompVis/stable-diffusion-v1-4",
|
infer.py
CHANGED
|
@@ -74,7 +74,7 @@ def get_pipe(
|
|
| 74 |
weight_name="ip-adapter-plus_sd14.bin",
|
| 75 |
)
|
| 76 |
|
| 77 |
-
elif
|
| 78 |
|
| 79 |
print('Pipe with ControlNet')
|
| 80 |
|
|
@@ -90,7 +90,7 @@ def get_pipe(
|
|
| 90 |
safety_checker=None,
|
| 91 |
).to(device)
|
| 92 |
|
| 93 |
-
elif
|
| 94 |
|
| 95 |
print('Pipe with IpAdapter')
|
| 96 |
|
|
@@ -105,7 +105,7 @@ def get_pipe(
|
|
| 105 |
subfolder="models",
|
| 106 |
weight_name="ip-adapter-plus_sd14.bin")
|
| 107 |
|
| 108 |
-
|
| 109 |
|
| 110 |
print('Pipe with only SD')
|
| 111 |
|
|
|
|
| 74 |
weight_name="ip-adapter-plus_sd14.bin",
|
| 75 |
)
|
| 76 |
|
| 77 |
+
elif use_controlnet and not use_ip_adapter:
|
| 78 |
|
| 79 |
print('Pipe with ControlNet')
|
| 80 |
|
|
|
|
| 90 |
safety_checker=None,
|
| 91 |
).to(device)
|
| 92 |
|
| 93 |
+
elif use_ip_adapter and not use_controlnetconto:
|
| 94 |
|
| 95 |
print('Pipe with IpAdapter')
|
| 96 |
|
|
|
|
| 105 |
subfolder="models",
|
| 106 |
weight_name="ip-adapter-plus_sd14.bin")
|
| 107 |
|
| 108 |
+
elif not use_controlnet and not use_ip_adapter:
|
| 109 |
|
| 110 |
print('Pipe with only SD')
|
| 111 |
|