Spaces:
Runtime error
Runtime error
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 |
|