File tree Expand file tree Collapse file tree 1 file changed +1
-2
lines changed
inference/generativeai/llm-workshop/lab10-falcon-40b-and-7b Expand file tree Collapse file tree 1 file changed +1
-2
lines changed Original file line number Diff line number Diff line change 199
199
"source" : [
200
200
" %%writefile ./code_falcon40b_deepspeed/requirements.txt\n " ,
201
201
" einops\n " ,
202
- " torch==2.0.1\n " ,
203
202
" git+https://github.com/lanking520/DeepSpeed.git@falcon"
204
203
]
205
204
},
330
329
"source" : [
331
330
" # inference_image_uri = f\" {account_id}.dkr.ecr.{region}.amazonaws.com/djl-ds:latest\"\n " ,
332
331
" 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 " ,
334
333
" )\n " ,
335
334
" print(f\" Image going to be used is ---- > {inference_image_uri}\" )"
336
335
]
You can’t perform that action at this time.
0 commit comments