File tree Expand file tree Collapse file tree 1 file changed +2
-2
lines changed
src/sagemaker/huggingface Expand file tree Collapse file tree 1 file changed +2
-2
lines changed Original file line number Diff line number Diff line change @@ -70,10 +70,10 @@ def __init__(
7070 ``image_uri`` is provided. The current supported version is ``4.6.1``.
7171 tensorflow_version (str): TensorFlow version you want to use for
7272 executing your model training code. Defaults to ``None``. Required unless
73- ``pytorch_version`` is provided. The current supported version is ``1.6.0 ``.
73+ ``pytorch_version`` is provided. The current supported version is ``2.4.1 ``.
7474 pytorch_version (str): PyTorch version you want to use for
7575 executing your model training code. Defaults to ``None``. Required unless
76- ``tensorflow_version`` is provided. The current supported version is ``2.4.1 ``.
76+ ``tensorflow_version`` is provided. The current supported versions are ``1.7.1`` and ``1.6.0 ``.
7777 source_dir (str): Path (absolute, relative or an S3 URI) to a directory
7878 with any other training source code dependencies aside from the entry
7979 point file (default: None). If ``source_dir`` is an S3 URI, it must
You can’t perform that action at this time.
0 commit comments