diff --git a/server/pyproject.toml b/server/pyproject.toml index 1babf749..7291f08f 100644 --- a/server/pyproject.toml +++ b/server/pyproject.toml @@ -54,5 +54,8 @@ priority = "explicit" markers = ["private: marks tests as requiring an admin hf token (deselect with '-m \"not private\"')"] [build-system] -requires = ["poetry-core>=1.0.0"] +requires = [ + "poetry-core>=1.0.0" + "torch>=2.0.1" +] build-backend = "poetry.core.masonry.api"