Skip to content

Commit 2d519b5

Browse files
shrestha-bikashBikash Shrestha
andauthored
chore: update dependency versions like transformers and torch (#4822)
Co-authored-by: Bikash Shrestha <[email protected]>
1 parent 8c0efd6 commit 2d519b5

File tree

2 files changed

+11
-11
lines changed

2 files changed

+11
-11
lines changed
Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -1,3 +1,3 @@
1-
transformers==4.2.1
2-
torch==1.13.1
1+
transformers==4.40.0
2+
torch==2.2.0
33
pandas

responsible_ai/sm-clarify_text_explainability_text_sentiment_analysis/sm-clarify_text_explainability_text_sentiment_analysis.ipynb

Lines changed: 9 additions & 9 deletions
Original file line numberDiff line numberDiff line change
@@ -77,8 +77,8 @@
7777
"1. Kernel: Python 3 (Data Science) kernel on SageMaker Studio or `conda_python3` kernel on notebook instances\n",
7878
"2. Instance type: Any GPU instance. Here, we use `ml.g4dn.xlarge`\n",
7979
"3. [SageMaker Python SDK](https://pypi.org/project/sagemaker/) version 2.70.0 or greater\n",
80-
"4. [Transformers](https://pypi.org/project/transformers/) > 4.6.1\n",
81-
"5. [Datasets](https://pypi.org/project/datasets/) > 1.6.2\n",
80+
"4. [Transformers](https://pypi.org/project/transformers/) >= 4.40.0\n",
81+
"5. [Datasets](https://pypi.org/project/datasets/) >= 2.18.0\n",
8282
"\n",
8383
"Let's start by installing the required packages."
8484
]
@@ -91,7 +91,7 @@
9191
},
9292
"outputs": [],
9393
"source": [
94-
"!pip install \"datasets[s3]==1.6.2\" \"transformers==4.6.1\" --upgrade --quiet"
94+
"!pip install \"datasets[s3]==2.18.0\" \"transformers==4.40.0\" --upgrade --quiet"
9595
]
9696
},
9797
{
@@ -115,7 +115,7 @@
115115
},
116116
"outputs": [],
117117
"source": [
118-
"!pip install \"torch==1.6\" --upgrade --quiet\n",
118+
"!pip install \"torch==2.2.0\" --upgrade --quiet\n",
119119
"!pip install captum --upgrade --quiet"
120120
]
121121
},
@@ -649,9 +649,9 @@
649649
" source_dir=\"scripts\",\n",
650650
" instance_type=\"ml.g4dn.xlarge\",\n",
651651
" instance_count=1,\n",
652-
" transformers_version=\"4.6.1\",\n",
653-
" pytorch_version=\"1.7.1\",\n",
654-
" py_version=\"py36\",\n",
652+
" transformers_version=\"4.36.0\",\n",
653+
" pytorch_version=\"2.1.0\",\n",
654+
" py_version=\"py310\",\n",
655655
" role=role,\n",
656656
" hyperparameters=hyperparameters,\n",
657657
" disable_profiler=True,\n",
@@ -730,8 +730,8 @@
730730
" name=model_name,\n",
731731
" model_data=zipped_model_path,\n",
732732
" role=get_execution_role(),\n",
733-
" framework_version=\"1.7.1\",\n",
734-
" py_version=\"py3\",\n",
733+
" framework_version=\"2.2.0\",\n",
734+
" py_version=\"py310\",\n",
735735
")\n",
736736
"predictor = model.deploy(\n",
737737
" initial_instance_count=1, instance_type=\"ml.g4dn.xlarge\", endpoint_name=endpoint_name\n",

0 commit comments

Comments
 (0)