Skip to content

Commit be56aa6

Browse files
author
Qing Lan
authored
Update falcon-40b-deepspeed.ipynb (#4275)
update with the latest changes
1 parent c5f4df4 commit be56aa6

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)