fix

autogluon · Jun 9, 2023 · 2d6de47 · 2d6de47
1 parent a6bba8e
commit 2d6de47
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/src/autogluon/cloud/scripts/sagemaker_scripts/multimodal_serve.py b/src/autogluon/cloud/scripts/sagemaker_scripts/multimodal_serve.py
@@ -63,6 +63,8 @@ def transform_fn(model, request_body, input_content_type, output_content_type="a
         payload = pickle.loads(buf)
         data = pd.read_parquet(BytesIO(payload["data"]))
         inference_kwargs = payload["inference_kwargs"]
+        if inference_kwargs is None:
+            inference_kwargs = {}
 
     elif input_content_type == "application/x-autogluon-npy":
         buf = bytes(request_body)