torchcpu2

Author: YH0223

pyproject.toml

@@ -25,7 +25,7 @@ dependencies = [
     "fastapi>=0.109.0",
     "uvicorn[standard]>=0.27.0",
     "transformers>=4.36.0",
-    "torch>=2.1.0+cpu",
+    "torch>=2.1.0",
     "pydantic>=2.5.0",
 ]