We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent ee550bf commit f16999eCopy full SHA for f16999e
src/huggingface_inference_toolkit/diffusers_utils.py
@@ -47,6 +47,13 @@ def __call__(
47
prompt,
48
**kwargs,
49
):
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
+
57
# TODO: add support for more images (Reason is correct output)
58
if "num_images_per_prompt" in kwargs:
59
kwargs.pop("num_images_per_prompt")
0 commit comments