bump minimal pytorch to 2.0

LoicGrobol · Feb 13, 2024 · 81ad7d8 · 81ad7d8
1 parent c4f90a0
commit 81ad7d8
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 1 deletion.
diff --git a/CHANGELOG.md b/CHANGELOG.md
@@ -9,6 +9,10 @@ The format is based on [Keep a Changelog](http://keepachangelog.com/) and this p
 ## [Unreleased]
 [Unreleased]: https://github.com/LoicGrobol/zeldarose/compare/v0.8.0...HEAD
 
+### Changed
+
+- Pytorch compatibility changed to `>= 2.0, < 2.3`
+
 ## [0.8.0] - 2023-10-06
 [0.8.0]: https://github.com/LoicGrobol/zeldarose/compare/v0.7.3...v0.8.0
 

diff --git a/pyproject.toml b/pyproject.toml
@@ -34,7 +34,7 @@ dependencies = [
     "torchmetrics >= 0.9, < 2.0",
     "tokenizers >= 0.10, < 0.16",
     "tomli",
-    "torch >= 1.12.0, < 2.3.0",
+    "torch >= 2.0, < 2.3",
     "transformers >= 4.19.0, < 5.0.0, != 4.23.0, != 4.23.1",
 ]