SeedVR2-3B

Paused

aducsdr commited on Sep 8

Commit

82be5f3

verified ·

1 Parent(s): d9459d3

Update requirements.txt

Files changed (1) hide show

requirements.txt CHANGED Viewed

@@ -1,6 +1,6 @@
 # Critical PyTorch dependencies must come first
-torch==2.4.0
-torchvision==0.19.0
 # Common
 einops                  # Tensor operations
@@ -46,4 +46,5 @@ isort                 # Import sorting
 pre-commit              # Pre-commit hooks
 expecttest             # Pytorch dist tests
 hypothesis           # Fix randomness
-av                   # Process audio and video

 # Critical PyTorch dependencies must come first
+torch==2.6.0
+torchvision==0.21.1
 # Common
 einops                  # Tensor operations
 pre-commit              # Pre-commit hooks
 expecttest             # Pytorch dist tests
 hypothesis           # Fix randomness
+av                   # Process audio and video
+https://github.com/Dao-AILab/flash-attention/releases/download/v2.7.4.post1/flash_attn-2.7.4.post1+cu12torch2.6cxx11abiFALSE-cp310-cp310-linux_x86_64.whl