Skip to content

Commit 2c36539

Browse files
documentation: add current Hugging Face supported versions (#2285)
Co-authored-by: Ahsan Khan <[email protected]>
1 parent 6d39762 commit 2c36539

File tree

1 file changed

+6
-9
lines changed

1 file changed

+6
-9
lines changed

src/sagemaker/huggingface/estimator.py

Lines changed: 6 additions & 9 deletions
Original file line numberDiff line numberDiff line change
@@ -57,25 +57,22 @@ def __init__(
5757
5858
Args:
5959
py_version (str): Python version you want to use for executing your model training
60-
code. Defaults to ``None``. Required unless ``image_uri`` is provided. List
61-
of supported versions:
62-
https://github.com/aws/sagemaker-python-sdk#huggingface-sagemaker-estimators
60+
code. Defaults to ``None``. Required unless ``image_uri`` is provided. If
61+
using PyTorch, the current supported version is ``py36``. If using TensorFlow,
62+
the current supported version is ``py37``.
6363
entry_point (str): Path (absolute or relative) to the Python source
6464
file which should be executed as the entry point to training.
6565
If ``source_dir`` is specified, then ``entry_point``
6666
must point to a file located at the root of ``source_dir``.
6767
transformers_version (str): Transformers version you want to use for
6868
executing your model training code. Defaults to ``None``. Required unless
69-
``image_uri`` is provided. List of supported versions:
70-
https://github.com/aws/sagemaker-python-sdk#huggingface-sagemaker-estimators.
69+
``image_uri`` is provided. The current supported version is ``4.4.2``.
7170
tensorflow_version (str): TensorFlow version you want to use for
7271
executing your model training code. Defaults to ``None``. Required unless
73-
``pytorch_version`` is provided. List of supported versions:
74-
https://github.com/aws/sagemaker-python-sdk#huggingface-sagemaker-estimators.
72+
``pytorch_version`` is provided. The current supported version is ``1.6.0``.
7573
pytorch_version (str): PyTorch version you want to use for
7674
executing your model training code. Defaults to ``None``. Required unless
77-
``tensorflow_version`` is provided. List of supported versions:
78-
https://github.com/aws/sagemaker-python-sdk#huggingface-sagemaker-estimators.
75+
``tensorflow_version`` is provided. The current supported version is ``2.4.1``.
7976
source_dir (str): Path (absolute, relative or an S3 URI) to a directory
8077
with any other training source code dependencies aside from the entry
8178
point file (default: None). If ``source_dir`` is an S3 URI, it must

0 commit comments

Comments
 (0)