Spaces:
Running
Running
Commit
•
43230a0
1
Parent(s):
ac592b3
appy.py: added controlnet of lineart.
Browse files
app.py
CHANGED
@@ -135,6 +135,11 @@ CONTROLNET_DICT = dict(
|
|
135 |
'strength': 1,
|
136 |
'preprocessor': 'canny',
|
137 |
},
|
|
|
|
|
|
|
|
|
|
|
138 |
)
|
139 |
|
140 |
last_check = 0
|
@@ -264,9 +269,7 @@ def generate_image (
|
|
264 |
num_steps,
|
265 |
identitynet_strength_ratio,
|
266 |
adapter_strength_ratio,
|
267 |
-
|
268 |
-
canny_strength,
|
269 |
-
depth_strength,
|
270 |
controlnet_selection,
|
271 |
guidance_scale,
|
272 |
seed,
|
@@ -305,9 +308,12 @@ def generate_image (
|
|
305 |
width = int(width * scaling)
|
306 |
height = int(height * scaling)
|
307 |
|
308 |
-
|
309 |
-
|
310 |
-
|
|
|
|
|
|
|
311 |
|
312 |
face_image_uploaded, ref_image_uploaded = upload_assets_with_cache(client, [face_image_path, ref_image_path])
|
313 |
|
@@ -492,6 +498,13 @@ with gr.Blocks(css=css) as demo:
|
|
492 |
step=0.05,
|
493 |
value=0.40,
|
494 |
)
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
495 |
with gr.Accordion(open=False, label='Advanced Options'):
|
496 |
negative_prompt = gr.Textbox(
|
497 |
label='Negative Prompt',
|
@@ -567,9 +580,12 @@ with gr.Blocks(css=css) as demo:
|
|
567 |
num_steps,
|
568 |
identitynet_strength_ratio,
|
569 |
adapter_strength_ratio,
|
570 |
-
|
571 |
-
|
572 |
-
|
|
|
|
|
|
|
573 |
controlnet_selection,
|
574 |
guidance_scale,
|
575 |
seed,
|
|
|
135 |
'strength': 1,
|
136 |
'preprocessor': 'canny',
|
137 |
},
|
138 |
+
lineart={
|
139 |
+
'model_name': 'controlnet-softedge-dexined-sdxl-1.0',
|
140 |
+
'strength': 1,
|
141 |
+
'preprocessor': 'lineart',
|
142 |
+
},
|
143 |
)
|
144 |
|
145 |
last_check = 0
|
|
|
269 |
num_steps,
|
270 |
identitynet_strength_ratio,
|
271 |
adapter_strength_ratio,
|
272 |
+
controlnet_strengths,
|
|
|
|
|
273 |
controlnet_selection,
|
274 |
guidance_scale,
|
275 |
seed,
|
|
|
308 |
width = int(width * scaling)
|
309 |
height = int(height * scaling)
|
310 |
|
311 |
+
(
|
312 |
+
CONTROLNET_DICT['pose']['strength'],
|
313 |
+
CONTROLNET_DICT['canny']['strength'],
|
314 |
+
CONTROLNET_DICT['depth']['strength'],
|
315 |
+
CONTROLNET_DICT['lineart']['strength'],
|
316 |
+
) = controlnet_strengths
|
317 |
|
318 |
face_image_uploaded, ref_image_uploaded = upload_assets_with_cache(client, [face_image_path, ref_image_path])
|
319 |
|
|
|
498 |
step=0.05,
|
499 |
value=0.40,
|
500 |
)
|
501 |
+
lineart_strength = gr.Slider(
|
502 |
+
label='Lineart strength',
|
503 |
+
minimum=0,
|
504 |
+
maximum=1.5,
|
505 |
+
step=0.05,
|
506 |
+
value=0.40,
|
507 |
+
)
|
508 |
with gr.Accordion(open=False, label='Advanced Options'):
|
509 |
negative_prompt = gr.Textbox(
|
510 |
label='Negative Prompt',
|
|
|
580 |
num_steps,
|
581 |
identitynet_strength_ratio,
|
582 |
adapter_strength_ratio,
|
583 |
+
[
|
584 |
+
pose_strength,
|
585 |
+
canny_strength,
|
586 |
+
depth_strength,
|
587 |
+
lineart_strength,
|
588 |
+
],
|
589 |
controlnet_selection,
|
590 |
guidance_scale,
|
591 |
seed,
|