diff --git a/pyproject.toml b/pyproject.toml
index 1ecd02ad..3dbf8570 100644
--- 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 = "*"
diff --git a/requirements.txt b/requirements.txt
index 2330d399..bd7c513c 100644
--- 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