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 @@ -69,10 +69,10 @@ def __init__(
69
69
``image_uri`` is provided. The current supported version is ``4.6.1``.
70
70
tensorflow_version (str): TensorFlow version you want to use for
71
71
executing your model training code. Defaults to ``None``. Required unless
72
- ``pytorch_version`` is provided. The current supported version is ``1.6.0 ``.
72
+ ``pytorch_version`` is provided. The current supported version is ``2.4.1 ``.
73
73
pytorch_version (str): PyTorch version you want to use for
74
74
executing your model training code. Defaults to ``None``. Required unless
75
- ``tensorflow_version`` is provided. The current supported version is ``2.4.1 ``.
75
+ ``tensorflow_version`` is provided. The current supported versions are ``1.7.1`` and ``1.6.0 ``.
76
76
source_dir (str): Path (absolute, relative or an S3 URI) to a directory
77
77
with any other training source code dependencies aside from the entry
78
78
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