Hugging Face's logo

Spaces:
nvidia
/
audio-flamingo-3
Running on A100

App Files Community
1
SreyanG-NVIDIA commited on
Commit
196e1f9
·
verified ·
1 Parent(s): b1c98a6

Update requirements.txt

Browse files
Files changed (1) hide show
  1. requirements.txt +2 -3
requirements.txt CHANGED
@@ -4,9 +4,8 @@ hydra-core
4
  loguru
5
  Pillow
6
  pydub
7
- torch==2.3.0+cu128
8
- torchvision==0.18.0+cu128
9
- --extra-index-url https://download.pytorch.org/whl/cu128
10
 
11
  # FlashAttention2
12
  flash_attn==2.7.3
 
4
  loguru
5
  Pillow
6
  pydub
7
+ torch
8
+ torchvision
 
9
 
10
  # FlashAttention2
11
  flash_attn==2.7.3