Skip to content

Commit 06e1f91

Browse files
Lokiiiiiiahsan-z-khan
authored andcommitted
feature: adding support for transformers 4.11 for SM Training Compiler
1 parent cf07ba7 commit 06e1f91

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

src/sagemaker/image_uri_config/huggingface-training-compiler.json

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -2,10 +2,10 @@
22
"training": {
33
"processors": ["gpu"],
44
"version_aliases": {
5-
"4.10": "4.10.2"
5+
"4.11": "4.11.0"
66
},
77
"versions": {
8-
"4.10.2": {
8+
"4.11.0": {
99
"version_aliases": {
1010
"pytorch1.9": "pytorch1.9.0",
1111
"tensorflow2.5": "tensorflow2.5.1"

0 commit comments

Comments
 (0)