SIGMitch commited on
Commit
dd0accd
·
verified ·
1 Parent(s): 7db26c5

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +5 -2
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": 1.0},
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,