Skip to content

Commit 9fad1c9

Browse files
authored
Merge branch 'main' into feature-processor-notebook
2 parents 73630b2 + be56aa6 commit 9fad1c9

File tree

1 file changed

+1
-2
lines changed

1 file changed

+1
-2
lines changed

inference/generativeai/llm-workshop/lab10-falcon-40b-and-7b/falcon-40b-deepspeed.ipynb

Lines changed: 1 addition & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -199,7 +199,6 @@
199199
"source": [
200200
"%%writefile ./code_falcon40b_deepspeed/requirements.txt\n",
201201
"einops\n",
202-
"torch==2.0.1\n",
203202
"git+https://github.com/lanking520/DeepSpeed.git@falcon"
204203
]
205204
},
@@ -330,7 +329,7 @@
330329
"source": [
331330
"# inference_image_uri = f\"{account_id}.dkr.ecr.{region}.amazonaws.com/djl-ds:latest\"\n",
332331
"inference_image_uri = (\n",
333-
" f\"763104351884.dkr.ecr.{region}.amazonaws.com/djl-inference:0.22.1-deepspeed0.8.3-cu118\"\n",
332+
" f\"763104351884.dkr.ecr.{region}.amazonaws.com/djl-inference:0.22.1-deepspeed0.9.2-cu118\"\n",
334333
")\n",
335334
"print(f\"Image going to be used is ---- > {inference_image_uri}\")"
336335
]

0 commit comments

Comments
 (0)