Skip to content

Commit c06ef53

Browse files
author
Xiong Zeng
committed
Update doc
1 parent 38c4f3d commit c06ef53

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

src/sagemaker/serve/builder/model_builder.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -145,7 +145,7 @@ class ModelBuilder(Triton, DJL, JumpStart, TGI, Transformers):
145145
to the model server). Possible values for this argument are
146146
``TORCHSERVE``, ``MMS``, ``TENSORFLOW_SERVING``, ``DJL_SERVING``,
147147
``TRITON``, and``TGI``.
148-
model_metadata (Optional[Dict[str, str]): Dictionary used to override the HuggingFace
148+
model_metadata (Optional[Dict[str, Any]): Dictionary used to override the HuggingFace
149149
model metadata. Currently ``HF_TASK`` is overridable.
150150
"""
151151

0 commit comments

Comments
 (0)