From 616a24347d933633b4b5f9e09377af89c9807d17 Mon Sep 17 00:00:00 2001 From: Santiago Martinez Date: Sat, 29 Apr 2023 15:33:40 +0100 Subject: [PATCH] fix: Updated torch version to avoid security issue --- pyproject.toml | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/pyproject.toml b/pyproject.toml index 69b1be5..6812dab 100644 --- a/pyproject.toml +++ b/pyproject.toml @@ -19,7 +19,7 @@ dependencies = [ "pandas", "scikit-learn", "scipy", - "torch<2", + "torch>=1.13.1,<2", "torchaudio", "torchvision", "click",