diff --git a/src/huggingface_hub/inference/_client.py b/src/huggingface_hub/inference/_client.py
index 10a761dd5f..98b951bd40 100644
--- a/src/huggingface_hub/inference/_client.py
+++ b/src/huggingface_hub/inference/_client.py
@@ -2424,8 +2424,8 @@ def text_to_image(
                 One prompt to guide what NOT to include in image generation.
             height (`int`, *optional*):
                 The height in pixels of the output image
-            width (`float`, *optional*):
-                The width in pixels of the image to generate.
+            width (`int`, *optional*):
+                The width in pixels of the output image
             num_inference_steps (`int`, *optional*):
                 The number of denoising steps. More denoising steps usually lead to a higher quality image at the
                 expense of slower inference.
diff --git a/src/huggingface_hub/inference/_generated/_async_client.py b/src/huggingface_hub/inference/_generated/_async_client.py
index ed430ee208..3c814aca91 100644
--- a/src/huggingface_hub/inference/_generated/_async_client.py
+++ b/src/huggingface_hub/inference/_generated/_async_client.py
@@ -2480,8 +2480,8 @@ async def text_to_image(
                 One prompt to guide what NOT to include in image generation.
             height (`int`, *optional*):
                 The height in pixels of the output image
-            width (`float`, *optional*):
-                The width in pixels of the image to generate.
+            width (`int`, *optional*):
+                The width in pixels of the output image
             num_inference_steps (`int`, *optional*):
                 The number of denoising steps. More denoising steps usually lead to a higher quality image at the
                 expense of slower inference.