revert torch for testing

This commit is contained in:
Adil Hafeez 2025-08-11 12:54:49 -07:00
parent 8794a3a35f
commit 2324f7fb29
No known key found for this signature in database
GPG key ID: 9B18EF7691369645
3 changed files with 351 additions and 382 deletions

View file

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