MohamedRashad commited on
Commit
5b9fc85
·
1 Parent(s): d65d459

Reduce inference steps in generate_item_image function and remove print statement

Browse files
Files changed (1) hide show
  1. app.py +1 -2
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=8,
92
  width=1024,
93
  height=1024,
94
  generator=torch.Generator("cpu").manual_seed(0),
@@ -106,7 +106,6 @@ def generate_item_image(object_t2i_prompt):
106
  # api_name="/infer"
107
  # )[0]
108
  # image = Image.open(img_path)
109
- print("Image generated")
110
  trial_id, processed_image = preprocess_pil_image(image)
111
  yield trial_id, processed_image
112
 
 
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=4,
92
  width=1024,
93
  height=1024,
94
  generator=torch.Generator("cpu").manual_seed(0),
 
106
  # api_name="/infer"
107
  # )[0]
108
  # image = Image.open(img_path)
 
109
  trial_id, processed_image = preprocess_pil_image(image)
110
  yield trial_id, processed_image
111