Skip to content

Commit f16999e

Browse files
committed
is seed is in kwarg, pass generator instead
1 parent ee550bf commit f16999e

File tree

1 file changed

+7
-0
lines changed

1 file changed

+7
-0
lines changed

src/huggingface_inference_toolkit/diffusers_utils.py

Lines changed: 7 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -47,6 +47,13 @@ def __call__(
4747
prompt,
4848
**kwargs,
4949
):
50+
# diffusers doesn't support seed but rather the generator kwarg
51+
if "seed" in kwargs:
52+
seed = int(kwargs["seed"])
53+
generator = torch.Generator().manual_seed(seed)
54+
kwargs["generator"] = generator
55+
kwargs.pop("seed")
56+
5057
# TODO: add support for more images (Reason is correct output)
5158
if "num_images_per_prompt" in kwargs:
5259
kwargs.pop("num_images_per_prompt")

0 commit comments

Comments
 (0)