|
77 | 77 | "1. Kernel: Python 3 (Data Science) kernel on SageMaker Studio or `conda_python3` kernel on notebook instances\n",
|
78 | 78 | "2. Instance type: Any GPU instance. Here, we use `ml.g4dn.xlarge`\n",
|
79 | 79 | "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", |
82 | 82 | "\n",
|
83 | 83 | "Let's start by installing the required packages."
|
84 | 84 | ]
|
|
91 | 91 | },
|
92 | 92 | "outputs": [],
|
93 | 93 | "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" |
95 | 95 | ]
|
96 | 96 | },
|
97 | 97 | {
|
|
115 | 115 | },
|
116 | 116 | "outputs": [],
|
117 | 117 | "source": [
|
118 |
| - "!pip install \"torch==1.6\" --upgrade --quiet\n", |
| 118 | + "!pip install \"torch==2.2.0\" --upgrade --quiet\n", |
119 | 119 | "!pip install captum --upgrade --quiet"
|
120 | 120 | ]
|
121 | 121 | },
|
|
649 | 649 | " source_dir=\"scripts\",\n",
|
650 | 650 | " instance_type=\"ml.g4dn.xlarge\",\n",
|
651 | 651 | " 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", |
655 | 655 | " role=role,\n",
|
656 | 656 | " hyperparameters=hyperparameters,\n",
|
657 | 657 | " disable_profiler=True,\n",
|
|
730 | 730 | " name=model_name,\n",
|
731 | 731 | " model_data=zipped_model_path,\n",
|
732 | 732 | " 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", |
735 | 735 | ")\n",
|
736 | 736 | "predictor = model.deploy(\n",
|
737 | 737 | " initial_instance_count=1, instance_type=\"ml.g4dn.xlarge\", endpoint_name=endpoint_name\n",
|
|
0 commit comments