Skip to content

Commit bd83555

Browse files
author
Xiong Zeng
committed
Update doc
1 parent 70c9d4f commit bd83555

File tree

1 file changed

+3
-3
lines changed

1 file changed

+3
-3
lines changed

src/sagemaker/serve/builder/model_builder.py

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -140,7 +140,7 @@ class ModelBuilder(Triton, DJL, JumpStart, TGI, Transformers):
140140
``TORCHSERVE``, ``MMS``, ``TENSORFLOW_SERVING``, ``DJL_SERVING``,
141141
``TRITON``, and``TGI``.
142142
model_metadata (Optional[Dict[str, str]): Dictionary used to override the HuggingFace
143-
model metadata.
143+
model metadata. Currently ``HF_TASK`` is overridable.
144144
"""
145145

146146
model_path: Optional[str] = field(
@@ -237,8 +237,8 @@ class ModelBuilder(Triton, DJL, JumpStart, TGI, Transformers):
237237
model_server: Optional[ModelServer] = field(
238238
default=None, metadata={"help": "Define the model server to deploy to."}
239239
)
240-
model_metadata: Optional[Dict[str, str]] = field(
241-
default=None, metadata={"help": "Define the model metadata to override"}
240+
model_metadata: Optional[Dict[str, Any]] = field(
241+
default=None, metadata={"help": "Define the model metadata to override, currently supports `HF_TASK`"}
242242
)
243243

244244
def _build_validations(self):

0 commit comments

Comments
 (0)