Spaces:
Running
on
Zero
Running
on
Zero
Commit
·
5b9fc85
1
Parent(s):
d65d459
Reduce inference steps in generate_item_image function and remove print statement
Browse files
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=
|
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 |
|