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 @@ -68,10 +68,10 @@ def __init__(
68
68
``image_uri`` is provided. The current supported version is ``4.4.2``.
69
69
tensorflow_version (str): TensorFlow version you want to use for
70
70
executing your model training code. Defaults to ``None``. Required unless
71
- ``pytorch_version`` is provided. The current supported version is ``1.6.0 ``.
71
+ ``pytorch_version`` is provided. The current supported version is ``2.4.1 ``.
72
72
pytorch_version (str): PyTorch version you want to use for
73
73
executing your model training code. Defaults to ``None``. Required unless
74
- ``tensorflow_version`` is provided. The current supported version is ``2.4.1 ``.
74
+ ``tensorflow_version`` is provided. The current supported version is ``1.6.0 ``.
75
75
py_version (str): Python version you want to use for executing your model training
76
76
code. Defaults to ``None``. Required unless ``image_uri`` is provided. If
77
77
using PyTorch, the current supported version is ``py36``. If using TensorFlow,
You can’t perform that action at this time.
0 commit comments