omer11a commited on
Commit
80df329
·
1 Parent(s): 61862e4

Increased optimization steps

Browse files
Files changed (1) hide show
  1. app.py +3 -2
app.py CHANGED
@@ -159,13 +159,14 @@ def inference(
159
  end_step_size=final_step_size,
160
  loss_stopping_value=loss_threshold,
161
  num_clusters_per_box=num_clusters_per_subject,
 
162
  )
163
 
164
  register_attention_editor_diffusers(model, editor)
165
  return model(prompts, latents=start_code, guidance_scale=classifier_free_guidance_scale).images
166
 
167
 
168
- @spaces.GPU(duration=300)
169
  def generate(
170
  prompt,
171
  subject_token_indices,
@@ -299,7 +300,7 @@ def main():
299
  with gr.Column():
300
  gr.HTML(ADVANCED_OPTION_DESCRIPTION)
301
  batch_size = gr.Slider(minimum=1, maximum=5, step=1, value=1, label="Number of samples (limited to one sample on current space)")
302
- num_guidance_steps = gr.Slider(minimum=5, maximum=20, step=1, value=5, label="Number of timesteps to perform guidance")
303
  init_step_size = gr.Slider(minimum=0, maximum=50, step=0.5, value=30, label="Initial step size")
304
  final_step_size = gr.Slider(minimum=0, maximum=20, step=0.5, value=15, label="Final step size")
305
  num_clusters_per_subject = gr.Slider(minimum=0, maximum=5, step=0.5, value=3, label="Number of clusters per subject")
 
159
  end_step_size=final_step_size,
160
  loss_stopping_value=loss_threshold,
161
  num_clusters_per_box=num_clusters_per_subject,
162
+ max_resolution=32,
163
  )
164
 
165
  register_attention_editor_diffusers(model, editor)
166
  return model(prompts, latents=start_code, guidance_scale=classifier_free_guidance_scale).images
167
 
168
 
169
+ @spaces.GPU(duration=400)
170
  def generate(
171
  prompt,
172
  subject_token_indices,
 
300
  with gr.Column():
301
  gr.HTML(ADVANCED_OPTION_DESCRIPTION)
302
  batch_size = gr.Slider(minimum=1, maximum=5, step=1, value=1, label="Number of samples (limited to one sample on current space)")
303
+ num_guidance_steps = gr.Slider(minimum=5, maximum=20, step=1, value=8, label="Number of timesteps to perform guidance")
304
  init_step_size = gr.Slider(minimum=0, maximum=50, step=0.5, value=30, label="Initial step size")
305
  final_step_size = gr.Slider(minimum=0, maximum=20, step=0.5, value=15, label="Final step size")
306
  num_clusters_per_subject = gr.Slider(minimum=0, maximum=5, step=0.5, value=3, label="Number of clusters per subject")