Skip to content

Commit 97241ee

Browse files
authored
Add prompt in image_to_image type (#2956)
1 parent 74f023b commit 97241ee

File tree

1 file changed

+2
-0
lines changed

1 file changed

+2
-0
lines changed

src/huggingface_hub/inference/_generated/types/image_to_image.py

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -30,6 +30,8 @@ class ImageToImageParameters(BaseInferenceType):
3030
"""For diffusion models. The number of denoising steps. More denoising steps usually lead to
3131
a higher quality image at the expense of slower inference.
3232
"""
33+
prompt: Optional[str] = None
34+
"""The text prompt to guide the image generation."""
3335
target_size: Optional[ImageToImageTargetSize] = None
3436
"""The size in pixel of the output image."""
3537

0 commit comments

Comments
 (0)