Fix key error in image retrieval payload data content

triton-inference-server · Nov 13, 2024 · e4e9757 · e4e9757
1 parent a886400
commit e4e9757
Showing 1 changed file with 6 additions and 2 deletions.
diff --git a/genai-perf/genai_perf/inputs/converters/image_retrieval_converter.py b/genai-perf/genai_perf/inputs/converters/image_retrieval_converter.py
@@ -45,7 +45,7 @@ def check_config(self, config: InputsConfig) -> None:
     def convert(
         self, generic_dataset: GenericDataset, config: InputsConfig
     ) -> Dict[Any, Any]:
-        request_body: Dict[str, Any] = {"input": []}
+        request_body: Dict[str, Any] = {"data": []}
 
         for file_data in generic_dataset.files_data.values():
             for index, row in enumerate(file_data.rows):
@@ -58,7 +58,11 @@ def _create_payload(
         self, row: DataRow, config: InputsConfig
     ) -> Dict[Any, Any]:
         content = self._retrieve_content(row, config)
-        return content
+
+        payload = {
+            "input": content
+        }
+        return payload
 
     def _retrieve_content(
         self, row: DataRow, config: InputsConfig