diff --git a/pyproject.toml b/pyproject.toml index ec08f58185..34b967a820 100644 --- a/pyproject.toml +++ b/pyproject.toml @@ -68,6 +68,9 @@ dependencies=[ "ruamel.yaml", "sympy", "tabulate", + "torch==2.5.0", + "torchaudio==2.5.0", + "torchvision==0.20.0", "typing-extensions", ]