diff --git a/pyproject.toml b/pyproject.toml index dcbae0b..1c530da 100644 --- a/pyproject.toml +++ b/pyproject.toml @@ -22,7 +22,7 @@ dependencies = [ "ragatouille>=0.0.8.post4", "scenedetect>=0.6.5.2", "torchaudio>=2.5.0", - "torch==2.5.1", + "torch==2.8.0", "torchmetrics>=1.6.0", "torchvision>=0.20.0", "tqdm<=4.66.5", diff --git a/requirements.txt b/requirements.txt index 0082f91..6da9986 100644 --- a/requirements.txt +++ b/requirements.txt @@ -198,7 +198,7 @@ threadpoolctl==3.5.0 tiktoken==0.8.0 timm==1.0.14 tokenizers==0.21.0 -torch==2.5.1 +torch==2.8.0 torchaudio==2.5.1 torchmetrics==1.6.1 torchvision==0.20.1