From 8bf0d899f8f5c1a478971c76c0d61af75d18ae7e Mon Sep 17 00:00:00 2001 From: erikkaum Date: Mon, 7 Oct 2024 17:19:16 +0200 Subject: [PATCH] add comment for clarity --- src/huggingface_inference_toolkit/diffusers_utils.py | 1 + 1 file changed, 1 insertion(+) diff --git a/src/huggingface_inference_toolkit/diffusers_utils.py b/src/huggingface_inference_toolkit/diffusers_utils.py index 431fd02e..70b683ab 100644 --- a/src/huggingface_inference_toolkit/diffusers_utils.py +++ b/src/huggingface_inference_toolkit/diffusers_utils.py @@ -48,6 +48,7 @@ def __call__( **kwargs, ): # diffusers doesn't support seed but rather the generator kwarg + # see: https://github.com/huggingface/api-inference-community/blob/8e577e2d60957959ba02f474b2913d84a9086b82/docker_images/diffusers/app/pipelines/text_to_image.py#L172-L176 if "seed" in kwargs: seed = int(kwargs["seed"]) generator = torch.Generator().manual_seed(seed)