From 8b207cb347ce6bf189184a8dbd7437824ffd7638 Mon Sep 17 00:00:00 2001 From: Dmitrii Kapitan Date: Fri, 17 Oct 2025 18:54:40 +0200 Subject: [PATCH] update dependencies to integrate torchcodec --- environment.yml | 9 +++++---- pyproject.toml | 1 + 2 files changed, 6 insertions(+), 4 deletions(-) diff --git a/environment.yml b/environment.yml index 2268e08..58cd20e 100644 --- a/environment.yml +++ b/environment.yml @@ -9,11 +9,12 @@ channels: dependencies: - python=3.11 - cudatoolkit=11.8 - - pytorch=2.3.1 + - pytorch=2.5.1 - pytorch-cuda=11.8 - - torchvision=0.18.1 - - torchaudio=2.3.1 + - torchvision=0.20.1 + - torchaudio=2.5.1 - faiss-gpu-raft=1.8.0 - ipykernel - jupyterlab - - jupyterlab_widgets \ No newline at end of file + - jupyterlab_widgets + - ffmpeg<8 \ No newline at end of file diff --git a/pyproject.toml b/pyproject.toml index 1ead16e..b672d06 100644 --- a/pyproject.toml +++ b/pyproject.toml @@ -49,6 +49,7 @@ dependencies = [ "torchvision", "tqdm", "transformers>=4.54", + "torchcodec<0.2", "webcolors", ]