Update app.py
Browse files
app.py
CHANGED
@@ -69,6 +69,7 @@ def process(
|
|
69 |
randomize_seed_checkbox: bool,
|
70 |
strength_slider: float,
|
71 |
num_inference_steps_slider: int,
|
|
|
72 |
progress=gr.Progress(track_tqdm=True)
|
73 |
):
|
74 |
if not input_text:
|
@@ -99,7 +100,7 @@ def process(
|
|
99 |
num_images_per_prompt =2,
|
100 |
strength=strength_slider,
|
101 |
generator=generator,
|
102 |
-
joint_attention_kwargs={"scale":
|
103 |
num_inference_steps=num_inference_steps_slider
|
104 |
)
|
105 |
print('INFERENCE DONE')
|
@@ -142,6 +143,7 @@ with gr.Blocks() as demo:
|
|
142 |
placeholder="Enter your prompt",
|
143 |
container=False,
|
144 |
)
|
|
|
145 |
submit_button_component = gr.Button(
|
146 |
value='Submit', variant='primary', scale=0)
|
147 |
|
@@ -194,7 +196,8 @@ with gr.Blocks() as demo:
|
|
194 |
seed_slicer_component,
|
195 |
randomize_seed_checkbox_component,
|
196 |
strength_slider_component,
|
197 |
-
num_inference_steps_slider_component
|
|
|
198 |
],
|
199 |
outputs=[
|
200 |
output_image_component,
|
|
|
69 |
randomize_seed_checkbox: bool,
|
70 |
strength_slider: float,
|
71 |
num_inference_steps_slider: int,
|
72 |
+
num_influence: float,
|
73 |
progress=gr.Progress(track_tqdm=True)
|
74 |
):
|
75 |
if not input_text:
|
|
|
100 |
num_images_per_prompt =2,
|
101 |
strength=strength_slider,
|
102 |
generator=generator,
|
103 |
+
joint_attention_kwargs={"scale": num_influence},
|
104 |
num_inference_steps=num_inference_steps_slider
|
105 |
)
|
106 |
print('INFERENCE DONE')
|
|
|
143 |
placeholder="Enter your prompt",
|
144 |
container=False,
|
145 |
)
|
146 |
+
lorasteps = gr.Slider(label="Influence", minimum=0, maximum=2, step=0.1, value=1)
|
147 |
submit_button_component = gr.Button(
|
148 |
value='Submit', variant='primary', scale=0)
|
149 |
|
|
|
196 |
seed_slicer_component,
|
197 |
randomize_seed_checkbox_component,
|
198 |
strength_slider_component,
|
199 |
+
num_inference_steps_slider_component,
|
200 |
+
lorasteps
|
201 |
],
|
202 |
outputs=[
|
203 |
output_image_component,
|