fix(ocr): install torchvision from CPU index alongside torch
torchvision installed from PyPI expects CUDA torch operator registrations. Installing from the CPU whl index ensures torchvision matches the CPU-only torch build. Fixes 'torchvision::nms does not exist' RuntimeError on startup. Co-Authored-By: Claude Sonnet 4.6 <noreply@anthropic.com>
This commit is contained in:
@@ -2,6 +2,7 @@ fastapi[standard]==0.115.6
|
||||
surya-ocr==0.17.1
|
||||
kraken==7.0
|
||||
torch==2.7.1
|
||||
torchvision==0.22.1
|
||||
transformers>=4.56.1,<5.0.0
|
||||
pillow>=10.2.0,<11.0.0
|
||||
pypdfium2==4.30.0
|
||||
|
||||
Reference in New Issue
Block a user