Skip to content

Commit 86ca70e

Browse files
author
Wang Napieralski
committed
fix image config
1 parent 0b4a46c commit 86ca70e

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

src/sagemaker/image_uri_config/huggingface.json

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -142,14 +142,14 @@
142142
},
143143

144144
"inference": {
145-
"processors": ["gpu"],
145+
"processors": ["gpu", "cpu"],
146146
"version_aliases": {
147147
"4.6": "4.6.1"
148148
},
149149
"versions": {
150150
"4.6.1": {
151151
"version_aliases": {
152-
"pytorch1.6": "pytorch1.7.1",
152+
"pytorch1.7": "pytorch1.7.1",
153153
"tensorflow2.4": "tensorflow2.4.1"
154154
},
155155
"pytorch1.7.1": {

0 commit comments

Comments
 (0)