From 1c652f123816bca550b166cbdfce04a3e4f3e97d Mon Sep 17 00:00:00 2001 From: pintaoz-aws Date: Tue, 1 Oct 2024 16:07:15 -0700 Subject: [PATCH] Pass kwargs to HuggingFaceModel.deploy() --- src/sagemaker/huggingface/model.py | 6 +----- 1 file changed, 1 insertion(+), 5 deletions(-) diff --git a/src/sagemaker/huggingface/model.py b/src/sagemaker/huggingface/model.py index 2bdea34d6f..46319474fa 100644 --- 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(