fix pytorch

Signed-off-by: sirutBuasai <sirutbuasai27@outlook.com>

Author: sirutBuasai

docs/src/data/pytorch-inference-arm64/2.6-cpu-sagemaker.yml

@@ -28,3 +28,4 @@ packages:
   torchdata: "0.10.1"
   torchserve: "0.12.0"
   openmpi: "4.1.7"
+  sagemaker_pytorch_inference: "2.0.25"

docs/src/data/pytorch-inference/2.6-cpu-sagemaker.yml

@@ -28,3 +28,4 @@ packages:
   torchdata: "0.10.1"
   torchserve: "0.12.0"
   openmpi: "4.1.7"
+  sagemaker_pytorch_inference: "2.0.25"

docs/src/data/pytorch-inference/2.6-gpu-sagemaker.yml

@@ -32,3 +32,4 @@ packages:
   cudnn: "9.1.0.70"
   nccl: "2.23.4"
   openmpi: "4.1.7"
+  sagemaker_pytorch_inference: "2.0.25"

docs/src/data/pytorch-training/2.6-cpu-ec2.yml

@@ -24,4 +24,5 @@ packages:
   pytorch: "2.6.0"
   torchvision: "0.21.0"
   torchaudio: "2.6.0"
+  torchtext: "0.18.0"
   torchdata: "0.10.1"

docs/src/data/pytorch-training/2.6-cpu-sagemaker.yml

@@ -24,4 +24,5 @@ packages:
   pytorch: "2.6.0"
   torchvision: "0.21.0"
   torchaudio: "2.6.0"
+  torchtext: "0.18.0"
   torchdata: "0.10.1"

docs/src/data/pytorch-training/2.6-gpu-ec2.yml

@@ -25,6 +25,7 @@ packages:
   pytorch: "2.6.0"
   torchvision: "0.21.0"
   torchaudio: "2.6.0"
+  torchtext: "0.18.0"
   torchdata: "0.10.1"
   cuda: "12.6.3"
   cudnn: "9.7.0.66"

docs/src/data/pytorch-training/2.6-gpu-sagemaker.yml

@@ -25,6 +25,7 @@ packages:
   pytorch: "2.6.0"
   torchvision: "0.21.0"
   torchaudio: "2.6.0"
+  torchtext: "0.18.0"
   torchdata: "0.10.1"
   cuda: "12.6.3"
   cudnn: "9.7.0.66"

docs/src/global.yml

@@ -102,6 +102,7 @@ display_names:
   flash_attn: "Flash Attention"
   neuronx_sdk: "NeuronX SDK"
   openmpi: "OpenMPI"
+  sagemaker_pytorch_inference: "SageMaker PyTorch Inference Toolkit"
 
 # =============================================================================
 # Framework Groups for Support Policy Consolidation