Pass kwargs to HuggingFaceModel.deploy()

aws · Oct 1, 2024 · 1c652f1 · 1c652f1
1 parent 66d5fdf
commit 1c652f1
Showing 1 changed file with 1 addition and 5 deletions.
diff --git a/src/sagemaker/huggingface/model.py b/src/sagemaker/huggingface/model.py
@@ -334,11 +334,7 @@ def deploy(
             container_startup_health_check_timeout=container_startup_health_check_timeout,
             inference_recommendation_id=inference_recommendation_id,
             explainer_config=explainer_config,
-            endpoint_logging=kwargs.get("endpoint_logging", False),
-            endpoint_type=kwargs.get("endpoint_type", None),
-            resources=kwargs.get("resources", None),
-            managed_instance_scaling=kwargs.get("managed_instance_scaling", None),
-            routing_config=kwargs.get("routing_config", None),
+            **kwargs,
         )
 
     def register(