diff --git a/pyproject.toml b/pyproject.toml index 9062c49..1a94e03 100644 --- a/pyproject.toml +++ b/pyproject.toml @@ -8,7 +8,7 @@ authors = [ dependencies = [ "uvicorn>=0.27.1", "litestar>=2.6.1", - "torch>=2.2.0", + "torch==2.2.0+cpu", "numpy>=1.26.4", "python-dotenv>=1.0.1", ]