Merge pull request #94 from VikParuchuri/dev

Downgrade torch
VikParuchuri · May 8, 2024 · 40a1043 · 40a1043
2 parents 180c37b + 1de7b9d
commit 40a1043
Show file tree

Hide file tree

Showing 2 changed files with 39 additions and 81 deletions.
diff --git a/poetry.lock b/poetry.lock
diff --git a/pyproject.toml b/pyproject.toml
@@ -22,7 +22,7 @@ include = [
 [tool.poetry.dependencies]
 python = ">=3.9,<3.13,!=3.9.7"
 transformers = "4.36.2"
-torch = "^2.1.2"
+torch = "2.2.2"
 pydantic = "^2.5.3"
 pydantic-settings = "^2.1.0"
 python-dotenv = "^1.0.0"