update dependencies to integrate torchcodec

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
+  - ffmpeg<8

pyproject.toml

@@ -49,6 +49,7 @@ dependencies = [
     "torchvision",
     "tqdm",
     "transformers>=4.54",
+    "torchcodec<0.2",
     "webcolors",
 ]