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