Browse Source

fix: torch version mismatch (#43)

AlpinDale 1 year ago
parent
commit
1e7d28f96f
2 changed files with 2 additions and 2 deletions
  1. 1 1
      pyproject.toml
  2. 1 1
      requirements.txt

+ 1 - 1
pyproject.toml

@@ -3,7 +3,7 @@ requires = [
     "ninja",
     "packaging",
     "setuptools",
-    "torch >= 2.0.0",
+    "torch ~= 2.0.0",
     "wheel",
 ]
 build-backend = "setuptools.build_meta"

+ 1 - 1
requirements.txt

@@ -3,7 +3,7 @@ psutil
 ray >= 2.5.1
 sentencepiece
 numpy
-torch >= 2.0.0
+torch ~= 2.0.0
 transformers >= 4.33.3
 uvicorn
 openai # for fastapi's openai proxy emulation