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 |
|