We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
1 parent 48205ad commit 7f36702Copy full SHA for 7f36702
src/sagemaker/serve/builder/tei_builder.py
@@ -85,7 +85,7 @@ def _set_to_tgi(self):
85
logger.warning(messaging)
86
self.model_server = ModelServer.TGI
87
88
- def _create_tei_model(self, **kwargs) -> Type[Model]:
+ def _create_tei_model(self, **kwargs) -> Type[Model]:
89
"""Placeholder docstring"""
90
if self.nb_instance_type and "instance_type" not in kwargs:
91
kwargs.update({"instance_type": self.nb_instance_type})
@@ -94,7 +94,7 @@ def _create_tei_model(self, **kwargs) -> Type[Model]:
94
self.image_uri = image_uris.retrieve(
95
"huggingface-tei",
96
image_scope="inference",
97
- instance_type=kwargs.get("instance_type")
+ instance_type=kwargs.get("instance_type"),
98
)
99
100
pysdk_model = HuggingFaceModel(
0 commit comments