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 })
@@ -94,7 +94,7 @@ def _create_tei_model(self, **kwargs) -> Type[Model]:
9494 self .image_uri = image_uris .retrieve (
9595 "huggingface-tei" ,
9696 image_scope = "inference" ,
97- instance_type = kwargs .get ("instance_type" )
97+ instance_type = kwargs .get ("instance_type" ),
9898 )
9999
100100 pysdk_model = HuggingFaceModel (
You can’t perform that action at this time.
0 commit comments