From fb8f5da1e6dee0dd4e42b65e44c6645d5357204e Mon Sep 17 00:00:00 2001 From: Ty Facey Date: Sat, 23 Dec 2023 00:28:58 -0500 Subject: [PATCH] Update pyproject.toml --- modules/streaming_pipeline/pyproject.toml | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/modules/streaming_pipeline/pyproject.toml b/modules/streaming_pipeline/pyproject.toml index b34eb6d7..a21a8164 100644 --- a/modules/streaming_pipeline/pyproject.toml +++ b/modules/streaming_pipeline/pyproject.toml @@ -17,7 +17,7 @@ sentencepiece = "^0.1.99" qdrant-client = "1.1.1" fire = "^0.5.0" transformers = "^4.33.1" -torch = {version = "2.0.1+cpu", source = "torch-cpu"} +torch = { platform = "linux", version = "2.0.1+cpu", source = "torch-cpu" } pyyaml = "6.0.1" [tool.poetry.group.dev.dependencies]