update torch==2.6.0 (#526)

This commit is contained in:
Adil Hafeez 2025-08-11 13:23:40 -07:00 committed by GitHub
parent 93ff4d7b1f
commit 202409cc9a
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
4 changed files with 365 additions and 353 deletions

View file

@ -12,7 +12,7 @@ packages = [
[tool.poetry.dependencies]
python = "^3.10"
fastapi = "0.115.0"
torch = "2.4.1"
torch = "2.6.0"
uvicorn = "0.31.0"
transformers = "^4.37.0"
accelerate = "^1.0.0"