Skip to content

Commit 25625a8

Browse files
author
multimodalart
committed
black
1 parent 9f65cb9 commit 25625a8

File tree

1 file changed

+3
-3
lines changed

1 file changed

+3
-3
lines changed

docker_images/diffusers/app/pipelines/text_to_image.py

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -160,20 +160,20 @@ def __call__(self, inputs: str, **kwargs) -> "Image.Image":
160160
def _process_req(self, inputs, **kwargs):
161161
# only one image per prompt is supported
162162
kwargs["num_images_per_prompt"] = 1
163-
163+
164164
if "num_inference_steps" not in kwargs:
165165
default_num_steps = os.getenv("DEFAULT_NUM_INFERENCE_STEPS")
166166
if default_num_steps:
167167
kwargs["num_inference_steps"] = int(default_num_steps)
168168
elif self.is_karras_compatible:
169169
kwargs["num_inference_steps"] = 20
170170
# Else, don't specify anything, leave the default behaviour
171-
171+
172172
if "seed" in kwargs:
173173
seed = int(kwargs["seed"])
174174
generator = torch.Generator().manual_seed(kwargs["seed"])
175175
kwargs["generator"] = generator
176176
kwargs.pop("seed")
177-
177+
178178
images = self.ldm(inputs, **kwargs)["images"]
179179
return images[0]

0 commit comments

Comments
 (0)