Commit
·
dde6e4c
1
Parent(s):
eff3d69
66
Browse files- requirements.txt +1 -2
requirements.txt
CHANGED
|
@@ -3,7 +3,7 @@
|
|
| 3 |
torch==2.2.1
|
| 4 |
torchvision==0.17.1
|
| 5 |
torchaudio==2.2.1
|
| 6 |
-
transformers==4.
|
| 7 |
sentence-transformers==2.7.0
|
| 8 |
diffusers==0.26.3
|
| 9 |
decord==0.6.0
|
|
@@ -15,4 +15,3 @@ scipy
|
|
| 15 |
numpy
|
| 16 |
tqdm
|
| 17 |
requests
|
| 18 |
-
git+https://github.com/Dao-AILab/flash-attention.git@v2.4.2#egg=flash-attn&subdirectory=csrc
|
|
|
|
| 3 |
torch==2.2.1
|
| 4 |
torchvision==0.17.1
|
| 5 |
torchaudio==2.2.1
|
| 6 |
+
transformers==4.34.1
|
| 7 |
sentence-transformers==2.7.0
|
| 8 |
diffusers==0.26.3
|
| 9 |
decord==0.6.0
|
|
|
|
| 15 |
numpy
|
| 16 |
tqdm
|
| 17 |
requests
|
|
|