(retriever) Bump torch to 2.10, torchvision to 0.25, and vLLM to 0.17#1762
(retriever) Bump torch to 2.10, torchvision to 0.25, and vLLM to 0.17#1762charlesbluca merged 11 commits intomainfrom
Conversation
c99b03c to
a3fce62
Compare
… conflicts Keep vllm==0.17.0 from branch with sys_platform=='linux' guard from upstream; retain new upstream deps (duckdb, neo4j, langchain-nvidia-ai-endpoints). Co-Authored-By: Claude Sonnet 4.6 <noreply@anthropic.com>
Greptile SummaryThis PR bumps core ML dependencies in the
|
| Filename | Overview |
|---|---|
| nemo_retriever/pyproject.toml | Bumps torch to ~=2.10.0, torchvision to >=0.25,<0.26, pins accelerate==1.12.0, and upgrades vllm to 0.17.0 with updated wheel URLs; changes are internally consistent |
| .github/workflows/huggingface-nightly.yml | Updates OCR nightly wheel build to use torch==2.10.0 and torchvision==0.25.0, matching the pyproject.toml version bump |
Flowchart
%%{init: {'theme': 'neutral'}}%%
flowchart TD
A[nemo-retriever local extra] --> B[torch ~=2.10.0\npytorch-cu130 index]
A --> C[torchvision >=0.25,<0.26\npytorch-cu130 index]
A --> D[accelerate ==1.12.0]
A --> E[vllm ==0.17.0\ndirect wheel URL cu130]
B --> F[CUDA 13.0]
C --> F
E --> F
G[huggingface-nightly.yml\nOCR wheel build] --> H[torch==2.10.0\ntorchvision==0.25.0\nbuild-time pins]
H --> F
Reviews (2): Last reviewed commit: "Build nemotron-ocr wheel against torch 2..." | Re-trigger Greptile
Co-Authored-By: Claude Sonnet 4.6 <noreply@anthropic.com>
Description
Bumps core ML dependency versions in the
localextra and updates the nemotron-ocr nightly wheel build to match:~=2.9.1→~=2.10.0>=0.24,<0.25→>=0.25,<0.26>=1.1.0→==1.12.00.16.0→0.17.0(adds VLM support)torch==2.10.0/torchvision==0.25.0(CUDA 13.0 unchanged)Checklist