poetry add torch@^2.1.0

This commit is contained in:
Cédric Deltheil 2023-10-10 11:50:27 +02:00 committed by Cédric Deltheil
parent 9f3d064d14
commit 5158187e96
2 changed files with 2 additions and 2 deletions

2
poetry.lock generated
View file

@ -3439,4 +3439,4 @@ training = ["bitsandbytes", "datasets", "loguru", "prodigyopt", "pydantic", "sci
[metadata]
lock-version = "2.0"
python-versions = ">=3.10,<3.12"
content-hash = "d37a0d9e9176073651aa23aad45b4006f28bffb4dbd45a43abaa687a4bfc93d3"
content-hash = "da05256f6a2f3cd39d2329a0136f929c3c7bf13b60f5d5a1cb21206b807cd15f"

View file

@ -12,7 +12,7 @@ packages = [{include = "refiners", from = "src"}]
[tool.poetry.dependencies]
python = ">=3.10,<3.12"
jaxtyping = "^0.2.14"
torch = "^2.0.0"
torch = "^2.1.0"
safetensors = "^0.3.0"
numpy = "^1.24.2"
pillow = ">=9.5.0"