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):
8585 logger .warning (messaging )
8686 self .model_server = ModelServer .TGI
8787
88- def _create_tei_model (self , ** kwargs ) -> Type [Model ]:
88+ def _create_tei_model (self , ** kwargs ) -> Type [Model ]:
8989 """Placeholder docstring"""
9090 if self .nb_instance_type and "instance_type" not in kwargs :
9191 kwargs .update ({"instance_type" : self .nb_instance_type })
@@ -95,7 +95,7 @@ def _create_tei_model(self, **kwargs) -> Type[Model]:
9595 "huggingface-tei" ,
9696 image_scope = "inference" ,
9797 instance_type = kwargs .get ("instance_type" ),
98- region = self .sagemaker_session .boto_region_name
98+ region = self .sagemaker_session .boto_region_name ,
9999 )
100100
101101 pysdk_model = HuggingFaceModel (
You can’t perform that action at this time.
0 commit comments