File tree Expand file tree Collapse file tree 1 file changed +2
-2
lines changed
src/sagemaker/serve/builder Expand file tree Collapse file tree 1 file changed +2
-2
lines changed Original file line number Diff line number Diff line change @@ -85,7 +85,7 @@ def _set_to_tgi(self):
85
85
logger .warning (messaging )
86
86
self .model_server = ModelServer .TGI
87
87
88
- def _create_tei_model (self , ** kwargs ) -> Type [Model ]:
88
+ def _create_tei_model (self , ** kwargs ) -> Type [Model ]:
89
89
"""Placeholder docstring"""
90
90
if self .nb_instance_type and "instance_type" not in kwargs :
91
91
kwargs .update ({"instance_type" : self .nb_instance_type })
@@ -95,7 +95,7 @@ def _create_tei_model(self, **kwargs) -> Type[Model]:
95
95
"huggingface-tei" ,
96
96
image_scope = "inference" ,
97
97
instance_type = kwargs .get ("instance_type" ),
98
- region = self .sagemaker_session .boto_region_name
98
+ region = self .sagemaker_session .boto_region_name ,
99
99
)
100
100
101
101
pysdk_model = HuggingFaceModel (
You can’t perform that action at this time.
0 commit comments