torch verisoning

1 year ago · d97de1c009
parent 6c7de98d71
commit d97de1c009
2 changed files with 2 additions and 2 deletions
--- a/pyproject.toml
+++ b/pyproject.toml
@ -23,6 +23,7 @@ classifiers = [

 [tool.poetry.dependencies]
 python = "^3.8.1"
+torch = "2.1.1"
 transformers = "*"
 openai = "0.28.0"
 langchain = "*"
@ -30,7 +31,6 @@ asyncio = "*"
 nest_asyncio = "*"
 einops = "*"
 google-generativeai = "*"
-torch = "*"
 langchain-experimental = "*"
 playwright = "*"
 duckduckgo-search = "*"
--- a/requirements.txt
+++ b/requirements.txt
@ -12,7 +12,7 @@ playwright
 wget==3.2
 simpleaichat
 httpx
-torch
+torch==2.1.1
 open_clip_torch
 ggl
 beautifulsoup4