Skip to content

Commit 6c37e52

Browse files
committed
Update onnx tutorials with transformers install instructions
Signed-off-by: Charlie Truong <chtruong@nvidia.com>
1 parent e8d9f45 commit 6c37e52

File tree

2 files changed

+2
-26
lines changed

2 files changed

+2
-26
lines changed

tutorials/onnx_tensorrt/embedding/llama_embedding.ipynb

Lines changed: 1 addition & 13 deletions
Original file line numberDiff line numberDiff line change
@@ -41,7 +41,7 @@
4141
"metadata": {},
4242
"outputs": [],
4343
"source": [
44-
"!pip install onnxruntime"
44+
"!pushd .. && uv pip install onnxruntime transformers==4.51.3 && popd"
4545
]
4646
},
4747
{
@@ -251,18 +251,6 @@
251251
"display_name": "Python 3 (ipykernel)",
252252
"language": "python",
253253
"name": "python3"
254-
},
255-
"language_info": {
256-
"codemirror_mode": {
257-
"name": "ipython",
258-
"version": 3
259-
},
260-
"file_extension": ".py",
261-
"mimetype": "text/x-python",
262-
"name": "python",
263-
"nbconvert_exporter": "python",
264-
"pygments_lexer": "ipython3",
265-
"version": "3.12.3"
266254
}
267255
},
268256
"nbformat": 4,

tutorials/onnx_tensorrt/reranker/llama_reranker.ipynb

Lines changed: 1 addition & 13 deletions
Original file line numberDiff line numberDiff line change
@@ -41,7 +41,7 @@
4141
"metadata": {},
4242
"outputs": [],
4343
"source": [
44-
"!pip install onnxruntime"
44+
"!pushd .. && uv pip install onnxruntime transformers==4.51.3 && popd"
4545
]
4646
},
4747
{
@@ -177,18 +177,6 @@
177177
"display_name": "Python 3 (ipykernel)",
178178
"language": "python",
179179
"name": "python3"
180-
},
181-
"language_info": {
182-
"codemirror_mode": {
183-
"name": "ipython",
184-
"version": 3
185-
},
186-
"file_extension": ".py",
187-
"mimetype": "text/x-python",
188-
"name": "python",
189-
"nbconvert_exporter": "python",
190-
"pygments_lexer": "ipython3",
191-
"version": "3.12.3"
192180
}
193181
},
194182
"nbformat": 4,

0 commit comments

Comments
 (0)