From cd6e4c17d14286c5f2605e2b8291435abd3cc9d5 Mon Sep 17 00:00:00 2001 From: haohanchen-yagao Date: Wed, 5 Oct 2022 15:10:20 -0700 Subject: [PATCH] minor fix --- .../pytorch/model_parallel/gpt2/requirements.txt | 2 +- .../pytorch/model_parallel/gpt2/smp-train-gpt-simple.ipynb | 2 +- 2 files changed, 2 insertions(+), 2 deletions(-) diff --git a/training/distributed_training/pytorch/model_parallel/gpt2/requirements.txt b/training/distributed_training/pytorch/model_parallel/gpt2/requirements.txt index ce5268410a..67d35f169d 100644 --- a/training/distributed_training/pytorch/model_parallel/gpt2/requirements.txt +++ b/training/distributed_training/pytorch/model_parallel/gpt2/requirements.txt @@ -4,6 +4,6 @@ sagemaker sagemaker-experiments scipy torchnet -transformers==4.17.0 +transformers==4.21.0 smdebug humanize diff --git a/training/distributed_training/pytorch/model_parallel/gpt2/smp-train-gpt-simple.ipynb b/training/distributed_training/pytorch/model_parallel/gpt2/smp-train-gpt-simple.ipynb index d0dac37bc9..e6b8621f23 100644 --- a/training/distributed_training/pytorch/model_parallel/gpt2/smp-train-gpt-simple.ipynb +++ b/training/distributed_training/pytorch/model_parallel/gpt2/smp-train-gpt-simple.ipynb @@ -78,7 +78,7 @@ "import boto3\n", "import sagemaker\n", "from sagemaker import get_execution_role\n", - "from sagemaker.huggingface import HuggingFace\n", + "from sagemaker.pytorch import PyTorch\n", "\n", "role = (\n", " get_execution_role()\n",