Skip to content

Commit 3f778ad

Browse files
committed
Address PR feedbacks
1 parent a81e0fa commit 3f778ad

File tree

1 file changed

+4
-0
lines changed

1 file changed

+4
-0
lines changed

src/sagemaker/serve/builder/transformers_builder.py

Lines changed: 4 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -38,6 +38,10 @@
3838
DEFAULT_TIMEOUT = 1800
3939

4040

41+
"""Retrieves images for different libraries - Pytorch, TensorFlow from HuggingFace hub
42+
"""
43+
44+
4145
# pylint: disable=W0108
4246
class Transformers(ABC):
4347
"""Transformers build logic with ModelBuilder()"""

0 commit comments

Comments
 (0)