sagemaker-core 2.1.1__py3-none-any.whl → 2.3.1__py3-none-any.whl

This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
Files changed (49) hide show
  1. sagemaker/__init__.py +2 -0
  2. sagemaker/core/common_utils.py +119 -1
  3. sagemaker/core/experiments/experiment.py +3 -0
  4. sagemaker/core/fw_utils.py +56 -12
  5. sagemaker/core/git_utils.py +66 -0
  6. sagemaker/core/helper/session_helper.py +22 -10
  7. sagemaker/core/image_retriever/image_retriever_utils.py +1 -3
  8. sagemaker/core/image_uri_config/huggingface-llm-neuronx.json +111 -1
  9. sagemaker/core/image_uri_config/huggingface-llm.json +110 -1
  10. sagemaker/core/image_uri_config/huggingface-neuronx.json +182 -6
  11. sagemaker/core/image_uri_config/huggingface-vllm-neuronx.json +38 -0
  12. sagemaker/core/image_uri_config/huggingface.json +151 -2
  13. sagemaker/core/image_uri_config/sagemaker-tritonserver.json +40 -0
  14. sagemaker/core/image_uri_config/sklearn.json +48 -0
  15. sagemaker/core/image_uri_config/xgboost.json +84 -0
  16. sagemaker/core/image_uris.py +9 -3
  17. sagemaker/core/iterators.py +11 -0
  18. sagemaker/core/jumpstart/models.py +2 -0
  19. sagemaker/core/jumpstart/region_config.json +8 -0
  20. sagemaker/core/local/data.py +10 -0
  21. sagemaker/core/local/utils.py +6 -5
  22. sagemaker/core/model_monitor/clarify_model_monitoring.py +2 -0
  23. sagemaker/core/model_registry.py +1 -1
  24. sagemaker/core/modules/configs.py +14 -1
  25. sagemaker/core/modules/train/container_drivers/common/utils.py +2 -10
  26. sagemaker/core/modules/train/sm_recipes/utils.py +1 -1
  27. sagemaker/core/processing.py +2 -0
  28. sagemaker/core/remote_function/client.py +31 -6
  29. sagemaker/core/remote_function/core/pipeline_variables.py +0 -6
  30. sagemaker/core/remote_function/core/serialization.py +16 -28
  31. sagemaker/core/remote_function/core/stored_function.py +8 -11
  32. sagemaker/core/remote_function/errors.py +1 -3
  33. sagemaker/core/remote_function/invoke_function.py +1 -6
  34. sagemaker/core/remote_function/job.py +2 -21
  35. sagemaker/core/telemetry/constants.py +6 -8
  36. sagemaker/core/telemetry/telemetry_logging.py +6 -5
  37. sagemaker/core/training/configs.py +16 -4
  38. sagemaker/core/workflow/utilities.py +10 -3
  39. {sagemaker_core-2.1.1.dist-info → sagemaker_core-2.3.1.dist-info}/METADATA +1 -1
  40. {sagemaker_core-2.1.1.dist-info → sagemaker_core-2.3.1.dist-info}/RECORD +43 -47
  41. sagemaker/core/huggingface/__init__.py +0 -29
  42. sagemaker/core/huggingface/llm_utils.py +0 -150
  43. sagemaker/core/huggingface/processing.py +0 -139
  44. sagemaker/core/huggingface/training_compiler/__init__.py +0 -0
  45. sagemaker/core/huggingface/training_compiler/config.py +0 -167
  46. sagemaker/core/image_uri_config/__init__.py +0 -13
  47. {sagemaker_core-2.1.1.dist-info → sagemaker_core-2.3.1.dist-info}/WHEEL +0 -0
  48. {sagemaker_core-2.1.1.dist-info → sagemaker_core-2.3.1.dist-info}/licenses/LICENSE +0 -0
  49. {sagemaker_core-2.1.1.dist-info → sagemaker_core-2.3.1.dist-info}/top_level.txt +0 -0
sagemaker/core/image_uri_config/huggingface-llm.json CHANGED
@@ -16,7 +16,8 @@
16
16
  "2.3": "2.3.1",
17
17
  "3.0": "3.0.1",
18
18
  "3.2": "3.2.3",
19
- "3.1": "3.1.1"
19
+ "3.1": "3.1.1",
20
+ "3.3": "3.3.6"
20
21
  },
21
22
  "versions": {
22
23
  "0.6.0": {
@@ -1152,6 +1153,114 @@
1152
1153
  "container_version": {
1153
1154
  "gpu": "cu124-ubuntu22.04"
1154
1155
  }
1156
+ },
1157
+ "3.3.4": {
1158
+ "py_versions": [
1159
+ "py311"
1160
+ ],
1161
+ "registries": {
1162
+ "af-south-1": "626614931356",
1163
+ "ap-east-1": "871362719292",
1164
+ "ap-east-2": "975050140332",
1165
+ "ap-northeast-1": "763104351884",
1166
+ "ap-northeast-2": "763104351884",
1167
+ "ap-northeast-3": "364406365360",
1168
+ "ap-south-1": "763104351884",
1169
+ "ap-south-2": "772153158452",
1170
+ "ap-southeast-1": "763104351884",
1171
+ "ap-southeast-2": "763104351884",
1172
+ "ap-southeast-3": "907027046896",
1173
+ "ap-southeast-4": "457447274322",
1174
+ "ap-southeast-5": "550225433462",
1175
+ "ap-southeast-6": "633930458069",
1176
+ "ap-southeast-7": "590183813437",
1177
+ "ca-central-1": "763104351884",
1178
+ "ca-west-1": "204538143572",
1179
+ "cn-north-1": "727897471807",
1180
+ "cn-northwest-1": "727897471807",
1181
+ "eu-central-1": "763104351884",
1182
+ "eu-central-2": "380420809688",
1183
+ "eu-north-1": "763104351884",
1184
+ "eu-south-1": "692866216735",
1185
+ "eu-south-2": "503227376785",
1186
+ "eu-west-1": "763104351884",
1187
+ "eu-west-2": "763104351884",
1188
+ "eu-west-3": "763104351884",
1189
+ "il-central-1": "780543022126",
1190
+ "me-central-1": "914824155844",
1191
+ "me-south-1": "217643126080",
1192
+ "mx-central-1": "637423239942",
1193
+ "sa-east-1": "763104351884",
1194
+ "us-east-1": "763104351884",
1195
+ "us-east-2": "763104351884",
1196
+ "us-gov-east-1": "446045086412",
1197
+ "us-gov-west-1": "442386744353",
1198
+ "us-iso-east-1": "886529160074",
1199
+ "us-isob-east-1": "094389454867",
1200
+ "us-isof-east-1": "303241398832",
1201
+ "us-isof-south-1": "454834333376",
1202
+ "us-west-1": "763104351884",
1203
+ "us-west-2": "763104351884"
1204
+ },
1205
+ "tag_prefix": "2.7.0-tgi3.3.4",
1206
+ "repository": "huggingface-pytorch-tgi-inference",
1207
+ "container_version": {
1208
+ "gpu": "cu124-ubuntu22.04"
1209
+ }
1210
+ },
1211
+ "3.3.6": {
1212
+ "py_versions": [
1213
+ "py311"
1214
+ ],
1215
+ "registries": {
1216
+ "af-south-1": "626614931356",
1217
+ "ap-east-1": "871362719292",
1218
+ "ap-east-2": "975050140332",
1219
+ "ap-northeast-1": "763104351884",
1220
+ "ap-northeast-2": "763104351884",
1221
+ "ap-northeast-3": "364406365360",
1222
+ "ap-south-1": "763104351884",
1223
+ "ap-south-2": "772153158452",
1224
+ "ap-southeast-1": "763104351884",
1225
+ "ap-southeast-2": "763104351884",
1226
+ "ap-southeast-3": "907027046896",
1227
+ "ap-southeast-4": "457447274322",
1228
+ "ap-southeast-5": "550225433462",
1229
+ "ap-southeast-6": "633930458069",
1230
+ "ap-southeast-7": "590183813437",
1231
+ "ca-central-1": "763104351884",
1232
+ "ca-west-1": "204538143572",
1233
+ "cn-north-1": "727897471807",
1234
+ "cn-northwest-1": "727897471807",
1235
+ "eu-central-1": "763104351884",
1236
+ "eu-central-2": "380420809688",
1237
+ "eu-north-1": "763104351884",
1238
+ "eu-south-1": "692866216735",
1239
+ "eu-south-2": "503227376785",
1240
+ "eu-west-1": "763104351884",
1241
+ "eu-west-2": "763104351884",
1242
+ "eu-west-3": "763104351884",
1243
+ "il-central-1": "780543022126",
1244
+ "me-central-1": "914824155844",
1245
+ "me-south-1": "217643126080",
1246
+ "mx-central-1": "637423239942",
1247
+ "sa-east-1": "763104351884",
1248
+ "us-east-1": "763104351884",
1249
+ "us-east-2": "763104351884",
1250
+ "us-gov-east-1": "446045086412",
1251
+ "us-gov-west-1": "442386744353",
1252
+ "us-iso-east-1": "886529160074",
1253
+ "us-isob-east-1": "094389454867",
1254
+ "us-isof-east-1": "303241398832",
1255
+ "us-isof-south-1": "454834333376",
1256
+ "us-west-1": "763104351884",
1257
+ "us-west-2": "763104351884"
1258
+ },
1259
+ "tag_prefix": "2.7.0-tgi3.3.6",
1260
+ "repository": "huggingface-pytorch-tgi-inference",
1261
+ "container_version": {
1262
+ "gpu": "cu124-ubuntu22.04"
1263
+ }
1155
1264
  }
1156
1265
  }
1157
1266
  }
sagemaker/core/image_uri_config/huggingface-neuronx.json CHANGED
@@ -8,7 +8,9 @@
8
8
  "4.34": "4.34.1",
9
9
  "4.36": "4.36.2",
10
10
  "4.43": "4.43.2",
11
- "4.48": "4.48.1"
11
+ "4.48": "4.48.1",
12
+ "4.51": "4.51.0",
13
+ "4.55": "4.55.4"
12
14
  },
13
15
  "versions": {
14
16
  "4.28.1": {
@@ -63,7 +65,7 @@
63
65
  "py_versions": [
64
66
  "py310"
65
67
  ],
66
- "repository": "huggingface-pytorch-inference-neuronx",
68
+ "repository": "huggingface-pytorch-training-neuronx",
67
69
  "registries": {
68
70
  "ap-east-2": "975050140332",
69
71
  "ap-northeast-1": "763104351884",
@@ -107,7 +109,7 @@
107
109
  "py_versions": [
108
110
  "py310"
109
111
  ],
110
- "repository": "huggingface-pytorch-inference-neuronx",
112
+ "repository": "huggingface-pytorch-training-neuronx",
111
113
  "registries": {
112
114
  "ap-east-2": "975050140332",
113
115
  "ap-northeast-1": "763104351884",
@@ -151,7 +153,7 @@
151
153
  "py_versions": [
152
154
  "py310"
153
155
  ],
154
- "repository": "huggingface-pytorch-inference-neuronx",
156
+ "repository": "huggingface-pytorch-training-neuronx",
155
157
  "registries": {
156
158
  "ap-northeast-1": "763104351884",
157
159
  "ap-south-1": "763104351884",
@@ -194,7 +196,7 @@
194
196
  "py_versions": [
195
197
  "py310"
196
198
  ],
197
- "repository": "huggingface-pytorch-inference-neuronx",
199
+ "repository": "huggingface-pytorch-training-neuronx",
198
200
  "registries": {
199
201
  "ap-northeast-1": "763104351884",
200
202
  "ap-south-1": "763104351884",
@@ -228,6 +230,92 @@
228
230
  "sdk2.20.0"
229
231
  ]
230
232
  }
233
+ },
234
+ "4.51.0": {
235
+ "version_aliases": {
236
+ "pytorch2.7": "pytorch2.7.0"
237
+ },
238
+ "pytorch2.7.0": {
239
+ "py_versions": [
240
+ "py310"
241
+ ],
242
+ "repository": "huggingface-pytorch-training-neuronx",
243
+ "registries": {
244
+ "ap-northeast-1": "763104351884",
245
+ "ap-south-1": "763104351884",
246
+ "ap-south-2": "772153158452",
247
+ "ap-southeast-1": "763104351884",
248
+ "ap-southeast-2": "763104351884",
249
+ "ap-southeast-4": "457447274322",
250
+ "ap-southeast-5": "550225433462",
251
+ "ap-southeast-7": "590183813437",
252
+ "cn-north-1": "727897471807",
253
+ "cn-northwest-1": "727897471807",
254
+ "eu-central-1": "763104351884",
255
+ "eu-central-2": "380420809688",
256
+ "eu-south-2": "503227376785",
257
+ "eu-west-1": "763104351884",
258
+ "eu-west-3": "763104351884",
259
+ "il-central-1": "780543022126",
260
+ "mx-central-1":"637423239942",
261
+ "sa-east-1": "763104351884",
262
+ "us-east-1": "763104351884",
263
+ "us-east-2": "763104351884",
264
+ "us-gov-east-1": "446045086412",
265
+ "us-gov-west-1": "442386744353",
266
+ "us-west-2": "763104351884",
267
+ "ca-west-1": "204538143572"
268
+ },
269
+ "container_version": {
270
+ "inf": "ubuntu22.04"
271
+ },
272
+ "sdk_versions": [
273
+ "sdk2.24.1"
274
+ ]
275
+ }
276
+ },
277
+ "4.55.4": {
278
+ "version_aliases": {
279
+ "pytorch2.7": "pytorch2.7.0"
280
+ },
281
+ "pytorch2.7.0": {
282
+ "py_versions": [
283
+ "py310"
284
+ ],
285
+ "repository": "huggingface-pytorch-training-neuronx",
286
+ "registries": {
287
+ "ap-northeast-1": "763104351884",
288
+ "ap-south-1": "763104351884",
289
+ "ap-south-2": "772153158452",
290
+ "ap-southeast-1": "763104351884",
291
+ "ap-southeast-2": "763104351884",
292
+ "ap-southeast-4": "457447274322",
293
+ "ap-southeast-5": "550225433462",
294
+ "ap-southeast-7": "590183813437",
295
+ "cn-north-1": "727897471807",
296
+ "cn-northwest-1": "727897471807",
297
+ "eu-central-1": "763104351884",
298
+ "eu-central-2": "380420809688",
299
+ "eu-south-2": "503227376785",
300
+ "eu-west-1": "763104351884",
301
+ "eu-west-3": "763104351884",
302
+ "il-central-1": "780543022126",
303
+ "mx-central-1":"637423239942",
304
+ "sa-east-1": "763104351884",
305
+ "us-east-1": "763104351884",
306
+ "us-east-2": "763104351884",
307
+ "us-gov-east-1": "446045086412",
308
+ "us-gov-west-1": "442386744353",
309
+ "us-west-2": "763104351884",
310
+ "ca-west-1": "204538143572"
311
+ },
312
+ "container_version": {
313
+ "inf": "ubuntu22.04"
314
+ },
315
+ "sdk_versions": [
316
+ "sdk2.26.0"
317
+ ]
318
+ }
231
319
  }
232
320
  }
233
321
  },
@@ -239,7 +327,9 @@
239
327
  "4.28": "4.28.1",
240
328
  "4.34": "4.34.1",
241
329
  "4.36": "4.36.2",
242
- "4.43": "4.43.2"
330
+ "4.43": "4.43.2",
331
+ "4.51": "4.51.3",
332
+ "4.55": "4.55.4"
243
333
  },
244
334
  "versions": {
245
335
  "4.28.1": {
@@ -504,6 +594,92 @@
504
594
  "sdk2.20.0"
505
595
  ]
506
596
  }
597
+ },
598
+ "4.51.3": {
599
+ "version_aliases": {
600
+ "pytorch2.7": "pytorch2.7.1"
601
+ },
602
+ "pytorch2.7.1": {
603
+ "py_versions": [
604
+ "py310"
605
+ ],
606
+ "repository": "huggingface-pytorch-inference-neuronx",
607
+ "registries": {
608
+ "ap-northeast-1": "763104351884",
609
+ "ap-south-1": "763104351884",
610
+ "ap-south-2": "772153158452",
611
+ "ap-southeast-1": "763104351884",
612
+ "ap-southeast-2": "763104351884",
613
+ "ap-southeast-4": "457447274322",
614
+ "ap-southeast-5": "550225433462",
615
+ "ap-southeast-7": "590183813437",
616
+ "cn-north-1": "727897471807",
617
+ "cn-northwest-1": "727897471807",
618
+ "eu-central-1": "763104351884",
619
+ "eu-central-2": "380420809688",
620
+ "eu-south-2": "503227376785",
621
+ "eu-west-1": "763104351884",
622
+ "eu-west-3": "763104351884",
623
+ "il-central-1": "780543022126",
624
+ "mx-central-1":"637423239942",
625
+ "sa-east-1": "763104351884",
626
+ "us-east-1": "763104351884",
627
+ "us-east-2": "763104351884",
628
+ "us-gov-east-1": "446045086412",
629
+ "us-gov-west-1": "442386744353",
630
+ "us-west-2": "763104351884",
631
+ "ca-west-1": "204538143572"
632
+ },
633
+ "container_version": {
634
+ "inf": "ubuntu22.04"
635
+ },
636
+ "sdk_versions": [
637
+ "sdk2.24.1"
638
+ ]
639
+ }
640
+ },
641
+ "4.55.4": {
642
+ "version_aliases": {
643
+ "pytorch2.7": "pytorch2.7.1"
644
+ },
645
+ "pytorch2.7.1": {
646
+ "py_versions": [
647
+ "py310"
648
+ ],
649
+ "repository": "huggingface-pytorch-inference-neuronx",
650
+ "registries": {
651
+ "ap-northeast-1": "763104351884",
652
+ "ap-south-1": "763104351884",
653
+ "ap-south-2": "772153158452",
654
+ "ap-southeast-1": "763104351884",
655
+ "ap-southeast-2": "763104351884",
656
+ "ap-southeast-4": "457447274322",
657
+ "ap-southeast-5": "550225433462",
658
+ "ap-southeast-7": "590183813437",
659
+ "cn-north-1": "727897471807",
660
+ "cn-northwest-1": "727897471807",
661
+ "eu-central-1": "763104351884",
662
+ "eu-central-2": "380420809688",
663
+ "eu-south-2": "503227376785",
664
+ "eu-west-1": "763104351884",
665
+ "eu-west-3": "763104351884",
666
+ "il-central-1": "780543022126",
667
+ "mx-central-1":"637423239942",
668
+ "sa-east-1": "763104351884",
669
+ "us-east-1": "763104351884",
670
+ "us-east-2": "763104351884",
671
+ "us-gov-east-1": "446045086412",
672
+ "us-gov-west-1": "442386744353",
673
+ "us-west-2": "763104351884",
674
+ "ca-west-1": "204538143572"
675
+ },
676
+ "container_version": {
677
+ "inf": "ubuntu22.04"
678
+ },
679
+ "sdk_versions": [
680
+ "sdk2.26.0"
681
+ ]
682
+ }
507
683
  }
508
684
  }
509
685
  }
sagemaker/core/image_uri_config/huggingface-vllm-neuronx.json ADDED
@@ -0,0 +1,38 @@
1
+ {
2
+ "inference": {
3
+ "processors": [
4
+ "inf2"
5
+ ],
6
+ "version_aliases": {
7
+ "0.4": "0.4.1"
8
+ },
9
+ "versions": {
10
+ "0.4.1": {
11
+ "py_versions": [
12
+ "py310"
13
+ ],
14
+ "registries": {
15
+ "ap-northeast-1": "763104351884",
16
+ "ap-south-1": "763104351884",
17
+ "ap-southeast-1": "763104351884",
18
+ "ap-southeast-2": "763104351884",
19
+ "eu-central-1": "763104351884",
20
+ "eu-west-1": "763104351884",
21
+ "eu-west-3": "763104351884",
22
+ "sa-east-1": "763104351884",
23
+ "us-east-1": "763104351884",
24
+ "us-east-2": "763104351884",
25
+ "us-west-2": "763104351884"
26
+ },
27
+ "tag_prefix": "0.10.2",
28
+ "repository": "huggingface-vllm-inference-neuronx",
29
+ "container_version": {
30
+ "inf2": "ubuntu22.04"
31
+ },
32
+ "sdk_versions": [
33
+ "sdk2.26.0"
34
+ ]
35
+ }
36
+ }
37
+ }
38
+ }
sagemaker/core/image_uri_config/huggingface.json CHANGED
@@ -16,7 +16,9 @@
16
16
  "4.36": "4.36.0",
17
17
  "4.46": "4.46.1",
18
18
  "4.48": "4.48.0",
19
- "4.49": "4.49.0"
19
+ "4.49": "4.49.0",
20
+ "4.55": "4.55.0",
21
+ "4.56": "4.56.2"
20
22
  },
21
23
  "versions": {
22
24
  "4.4.2": {
@@ -1162,6 +1164,100 @@
1162
1164
  "gpu": "cu124-ubuntu22.04"
1163
1165
  }
1164
1166
  }
1167
+ },
1168
+ "4.55.0": {
1169
+ "version_aliases": {
1170
+ "pytorch2.7": "pytorch2.7.1"
1171
+ },
1172
+ "pytorch2.7.1": {
1173
+ "py_versions": [
1174
+ "py312"
1175
+ ],
1176
+ "registries": {
1177
+ "af-south-1": "626614931356",
1178
+ "il-central-1": "780543022126",
1179
+ "ap-east-1": "871362719292",
1180
+ "ap-northeast-1": "763104351884",
1181
+ "ap-northeast-2": "763104351884",
1182
+ "ap-northeast-3": "364406365360",
1183
+ "ap-south-1": "763104351884",
1184
+ "ap-southeast-1": "763104351884",
1185
+ "ap-southeast-2": "763104351884",
1186
+ "ap-southeast-3": "907027046896",
1187
+ "ca-central-1": "763104351884",
1188
+ "cn-north-1": "727897471807",
1189
+ "cn-northwest-1": "727897471807",
1190
+ "eu-central-1": "763104351884",
1191
+ "eu-north-1": "763104351884",
1192
+ "eu-west-1": "763104351884",
1193
+ "eu-west-2": "763104351884",
1194
+ "eu-west-3": "763104351884",
1195
+ "eu-south-1": "692866216735",
1196
+ "me-south-1": "217643126080",
1197
+ "me-central-1": "914824155844",
1198
+ "sa-east-1": "763104351884",
1199
+ "us-east-1": "763104351884",
1200
+ "us-east-2": "763104351884",
1201
+ "us-gov-east-1": "446045086412",
1202
+ "us-gov-west-1": "442386744353",
1203
+ "us-iso-east-1": "886529160074",
1204
+ "us-isob-east-1": "094389454867",
1205
+ "us-west-1": "763104351884",
1206
+ "us-west-2": "763104351884",
1207
+ "ca-west-1": "204538143572"
1208
+ },
1209
+ "repository": "huggingface-pytorch-training",
1210
+ "container_version": {
1211
+ "gpu": "cu128-ubuntu22.04"
1212
+ }
1213
+ }
1214
+ },
1215
+ "4.56.2": {
1216
+ "version_aliases": {
1217
+ "pytorch2.8": "pytorch2.8.0"
1218
+ },
1219
+ "pytorch2.8.0": {
1220
+ "py_versions": [
1221
+ "py312"
1222
+ ],
1223
+ "registries": {
1224
+ "af-south-1": "626614931356",
1225
+ "il-central-1": "780543022126",
1226
+ "ap-east-1": "871362719292",
1227
+ "ap-northeast-1": "763104351884",
1228
+ "ap-northeast-2": "763104351884",
1229
+ "ap-northeast-3": "364406365360",
1230
+ "ap-south-1": "763104351884",
1231
+ "ap-southeast-1": "763104351884",
1232
+ "ap-southeast-2": "763104351884",
1233
+ "ap-southeast-3": "907027046896",
1234
+ "ca-central-1": "763104351884",
1235
+ "cn-north-1": "727897471807",
1236
+ "cn-northwest-1": "727897471807",
1237
+ "eu-central-1": "763104351884",
1238
+ "eu-north-1": "763104351884",
1239
+ "eu-west-1": "763104351884",
1240
+ "eu-west-2": "763104351884",
1241
+ "eu-west-3": "763104351884",
1242
+ "eu-south-1": "692866216735",
1243
+ "me-south-1": "217643126080",
1244
+ "me-central-1": "914824155844",
1245
+ "sa-east-1": "763104351884",
1246
+ "us-east-1": "763104351884",
1247
+ "us-east-2": "763104351884",
1248
+ "us-gov-east-1": "446045086412",
1249
+ "us-gov-west-1": "442386744353",
1250
+ "us-iso-east-1": "886529160074",
1251
+ "us-isob-east-1": "094389454867",
1252
+ "us-west-1": "763104351884",
1253
+ "us-west-2": "763104351884",
1254
+ "ca-west-1": "204538143572"
1255
+ },
1256
+ "repository": "huggingface-pytorch-training",
1257
+ "container_version": {
1258
+ "gpu": "cu129-ubuntu22.04"
1259
+ }
1260
+ }
1165
1261
  }
1166
1262
  }
1167
1263
  },
@@ -1179,7 +1275,8 @@
1179
1275
  "4.26": "4.26.0",
1180
1276
  "4.28": "4.28.1",
1181
1277
  "4.37": "4.37.0",
1182
- "4.49": "4.49.0"
1278
+ "4.49": "4.49.0",
1279
+ "4.51": "4.51.3"
1183
1280
  },
1184
1281
  "versions": {
1185
1282
  "4.6.1": {
@@ -2132,6 +2229,58 @@
2132
2229
  "cpu": "ubuntu22.04"
2133
2230
  }
2134
2231
  }
2232
+ },
2233
+ "4.51.3": {
2234
+ "version_aliases": {
2235
+ "pytorch2.6": "pytorch2.6.0"
2236
+ },
2237
+ "pytorch2.6.0": {
2238
+ "py_versions": [
2239
+ "py312"
2240
+ ],
2241
+ "registries": {
2242
+ "af-south-1": "626614931356",
2243
+ "il-central-1": "780543022126",
2244
+ "ap-east-1": "871362719292",
2245
+ "ap-northeast-1": "763104351884",
2246
+ "ap-northeast-2": "763104351884",
2247
+ "ap-northeast-3": "364406365360",
2248
+ "ap-south-1": "763104351884",
2249
+ "ap-south-2": "772153158452",
2250
+ "ap-southeast-1": "763104351884",
2251
+ "ap-southeast-2": "763104351884",
2252
+ "ap-southeast-3": "907027046896",
2253
+ "ap-southeast-4": "457447274322",
2254
+ "ca-central-1": "763104351884",
2255
+ "cn-north-1": "727897471807",
2256
+ "cn-northwest-1": "727897471807",
2257
+ "eu-central-1": "763104351884",
2258
+ "eu-central-2": "380420809688",
2259
+ "eu-north-1": "763104351884",
2260
+ "eu-west-1": "763104351884",
2261
+ "eu-west-2": "763104351884",
2262
+ "eu-west-3": "763104351884",
2263
+ "eu-south-1": "692866216735",
2264
+ "eu-south-2": "503227376785",
2265
+ "me-south-1": "217643126080",
2266
+ "me-central-1": "914824155844",
2267
+ "sa-east-1": "763104351884",
2268
+ "us-east-1": "763104351884",
2269
+ "us-east-2": "763104351884",
2270
+ "us-gov-east-1": "446045086412",
2271
+ "us-gov-west-1": "442386744353",
2272
+ "us-iso-east-1": "886529160074",
2273
+ "us-isob-east-1": "094389454867",
2274
+ "us-west-1": "763104351884",
2275
+ "us-west-2": "763104351884",
2276
+ "ca-west-1": "204538143572"
2277
+ },
2278
+ "repository": "huggingface-pytorch-inference",
2279
+ "container_version": {
2280
+ "gpu": "cu124-ubuntu22.04",
2281
+ "cpu": "ubuntu22.04"
2282
+ }
2283
+ }
2135
2284
  }
2136
2285
  }
2137
2286
  }
sagemaker/core/image_uri_config/sagemaker-tritonserver.json CHANGED
@@ -7,6 +7,46 @@
7
7
  "inference"
8
8
  ],
9
9
  "versions": {
10
+ "25.09": {
11
+ "registries": {
12
+ "af-south-1": "626614931356",
13
+ "ap-east-1": "871362719292",
14
+ "ap-east-2": "975050140332",
15
+ "ap-northeast-1": "763104351884",
16
+ "ap-northeast-2": "763104351884",
17
+ "ap-northeast-3": "364406365360",
18
+ "ap-south-1": "763104351884",
19
+ "ap-south-2": "772153158452",
20
+ "ap-southeast-1": "763104351884",
21
+ "ap-southeast-2": "763104351884",
22
+ "ap-southeast-3": "907027046896",
23
+ "ap-southeast-4": "457447274322",
24
+ "ap-southeast-5": "550225433462",
25
+ "ap-southeast-6": "633930458069",
26
+ "ap-southeast-7": "590183813437",
27
+ "ca-central-1": "763104351884",
28
+ "ca-west-1": "204538143572",
29
+ "eu-central-1": "763104351884",
30
+ "eu-central-2": "380420809688",
31
+ "eu-north-1": "763104351884",
32
+ "eu-south-1": "692866216735",
33
+ "eu-south-2": "503227376785",
34
+ "eu-west-1": "763104351884",
35
+ "eu-west-2": "763104351884",
36
+ "eu-west-3": "763104351884",
37
+ "il-central-1": "780543022126",
38
+ "me-central-1": "914824155844",
39
+ "me-south-1": "217643126080",
40
+ "mx-central-1": "637423239942",
41
+ "sa-east-1": "763104351884",
42
+ "us-east-1": "763104351884",
43
+ "us-east-2": "763104351884",
44
+ "us-west-1": "763104351884",
45
+ "us-west-2": "763104351884"
46
+ },
47
+ "repository": "sagemaker-tritonserver",
48
+ "tag_prefix": "25.09-py3"
49
+ },
10
50
  "25.04": {
11
51
  "registries": {
12
52
  "af-south-1": "626614931356",