diff --git a/pyproject.toml b/pyproject.toml
index 1612e39..5f63b6a 100644
--- a/pyproject.toml
+++ b/pyproject.toml
@@ -12,7 +12,7 @@ urls = {homepage = "https://github.com/deepseek-ai/DeepSeek-VL2"}
 readme = "README.md"
 requires-python = ">=3.8"
 dependencies = [
-    "torch>=2.0.1",
+    "torch==2.0.1",
     "transformers==4.38.2",
     "timm>=0.9.16",
     "accelerate",