diff --git a/src/huggingface_inference_toolkit/diffusers_utils.py b/src/huggingface_inference_toolkit/diffusers_utils.py index 54fad5ad..431fd02e 100644 --- a/src/huggingface_inference_toolkit/diffusers_utils.py +++ b/src/huggingface_inference_toolkit/diffusers_utils.py @@ -47,6 +47,13 @@ def __call__( prompt, **kwargs, ): + # diffusers doesn't support seed but rather the generator kwarg + if "seed" in kwargs: + seed = int(kwargs["seed"]) + generator = torch.Generator().manual_seed(seed) + kwargs["generator"] = generator + kwargs.pop("seed") + # TODO: add support for more images (Reason is correct output) if "num_images_per_prompt" in kwargs: kwargs.pop("num_images_per_prompt")