Fix Replicate image-to-image compatibility with different model schemas (#3749)

Pass the input image in multiple payload keys (image, images, input_image,
input_images) to support different Replicate models that expect the image
in different keys.

Similar to: huggingface/huggingface.js#1938

Co-authored-by: Cursor Agent <cursoragent@cursor.com>

Author: hanouticelina

src/huggingface_hub/inference/_providers/replicate.py

@@ -142,7 +142,16 @@ def _prepare_payload_as_dict(
     ) -> Optional[dict]:
         image_url = _as_url(inputs, default_mime_type="image/jpeg")
 
-        payload: dict[str, Any] = {"input": {"input_image": image_url, **filter_none(parameters)}}
+        # Different Replicate models expect the image in different keys
+        payload: dict[str, Any] = {
+            "input": {
+                "image": image_url,
+                "images": [image_url],
+                "input_image": image_url,
+                "input_images": [image_url],
+                **filter_none(parameters),
+            }
+        }
 
         mapped_model = provider_mapping_info.provider_id
         if ":" in mapped_model:

tests/test_inference_providers.py

@@ -1651,7 +1651,13 @@ def test_image_to_image_payload(self):
             ),
         )
         assert payload == {
-            "input": {"input_image": image_uri, "num_inference_steps": 20},
+            "input": {
+                "image": image_uri,
+                "images": [image_uri],
+                "input_image": image_uri,
+                "input_images": [image_uri],
+                "num_inference_steps": 20,
+            },
         }
 
         payload = helper._prepare_payload_as_dict(
@@ -1666,7 +1672,13 @@ def test_image_to_image_payload(self):
             ),
         )
         assert payload == {
-            "input": {"input_image": image_uri, "num_inference_steps": 20},
+            "input": {
+                "image": image_uri,
+                "images": [image_uri],
+                "input_image": image_uri,
+                "input_images": [image_uri],
+                "num_inference_steps": 20,
+            },
             "version": "123456",
         }