update torch==2.6.0

This commit is contained in:
Adil Hafeez 2025-07-11 23:09:18 -07:00
parent 14f19f02a7
commit f728a1a975
No known key found for this signature in database
GPG key ID: 9B18EF7691369645
14 changed files with 124 additions and 102 deletions

View file

@ -1,6 +1,6 @@
[tool.poetry]
name = "archgw_modelserver"
version = "0.3.4"
version = "0.3.5"
description = "A model server for serving models"
authors = ["Katanemo Labs, Inc <info@katanemo.com>"]
license = "Apache 2.0"
@ -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"