RishubhPar commited on
Commit
fcdef42
·
verified ·
1 Parent(s): 70c90ad

changed number of inference steps

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -424,7 +424,7 @@ def generate_image_stack_edits(text_prompt, n_edits, input_image):
424
  image=img,
425
  height=img.height,
426
  width=img.width,
427
- num_inference_steps=6, # 28
428
  prompt_embeds=pe,
429
  pooled_prompt_embeds=ppe,
430
  generator=gen,
@@ -458,7 +458,7 @@ def generate_single_image(text_prompt, slider_value, input_image):
458
  image=img,
459
  height=img.height,
460
  width=img.width,
461
- num_inference_steps=6, # 28
462
  prompt_embeds=pe,
463
  pooled_prompt_embeds=ppe,
464
  generator=gen,
 
424
  image=img,
425
  height=img.height,
426
  width=img.width,
427
+ num_inference_steps=14, # 28
428
  prompt_embeds=pe,
429
  pooled_prompt_embeds=ppe,
430
  generator=gen,
 
458
  image=img,
459
  height=img.height,
460
  width=img.width,
461
+ num_inference_steps=14, # 28
462
  prompt_embeds=pe,
463
  pooled_prompt_embeds=ppe,
464
  generator=gen,