MohamedRashad commited on
Commit
2c55a9a
·
1 Parent(s): 196afc8

Reduce inference steps to 1 in generate_item_image function

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -88,7 +88,7 @@ def generate_item_image(object_t2i_prompt):
88
  for image in pipe.flux_pipe_call_that_returns_an_iterable_of_images(
89
  prompt=object_t2i_prompt,
90
  guidance_scale=3.5,
91
- num_inference_steps=28,
92
  width=512,
93
  height=512,
94
  generator=torch.Generator("cpu").manual_seed(0),
 
88
  for image in pipe.flux_pipe_call_that_returns_an_iterable_of_images(
89
  prompt=object_t2i_prompt,
90
  guidance_scale=3.5,
91
+ num_inference_steps=1,
92
  width=512,
93
  height=512,
94
  generator=torch.Generator("cpu").manual_seed(0),