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__(
70
70
``image_uri`` is provided. The current supported version is ``4.6.1``.
71
71
tensorflow_version (str): TensorFlow version you want to use for
72
72
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 ``.
74
74
pytorch_version (str): PyTorch version you want to use for
75
75
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 ``.
77
77
source_dir (str): Path (absolute, relative or an S3 URI) to a directory
78
78
with any other training source code dependencies aside from the entry
79
79
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