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",