downgrade torch version to ^2.0.0 in pyproject.toml to fix the memory…

… issues.
JohnSnowLabs · Jan 3, 2025 · 1556b60 · 1556b60
1 parent 770d632
commit 1556b60
Show file tree

Hide file tree

Showing 2 changed files with 31 additions and 215 deletions.
diff --git a/poetry.lock b/poetry.lock
diff --git a/pyproject.toml b/pyproject.toml
@@ -61,7 +61,7 @@ spacy = { version = ">=3.0.0", optional = true }
 nest-asyncio = "^1.5.0"
 openai = "^1.40.6"
 jsonlines = ">=3.1.0"
-torch = {version = "2.2.0", optional = true}
+torch = { version = "^2.0.0", optional = true }
 pandas = "^2.0.3"
 pyyaml = "^6.0"
 tqdm = "^4.67.1"