Spaces:
Running
on
Zero
Running
on
Zero
Update requirements.txt
Browse files- requirements.txt +27 -6
requirements.txt
CHANGED
@@ -1,6 +1,27 @@
|
|
1 |
-
|
2 |
-
|
3 |
-
|
4 |
-
|
5 |
-
|
6 |
-
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
1 |
+
pytest
|
2 |
+
matplotlib
|
3 |
+
scikit-learn
|
4 |
+
scipy
|
5 |
+
spacy
|
6 |
+
numpy
|
7 |
+
einops
|
8 |
+
einsum
|
9 |
+
fvcore
|
10 |
+
h5py
|
11 |
+
twine
|
12 |
+
transformers==4.45.2
|
13 |
+
huggingface_hub==0.24
|
14 |
+
accelerate==0.34.2
|
15 |
+
diffusers==0.30.3
|
16 |
+
pillow==10.2.0
|
17 |
+
torch==2.3.1
|
18 |
+
torchvision==0.18.1
|
19 |
+
torchaudio==2.3.1
|
20 |
+
flash-attn==2.6.3
|
21 |
+
git+https://github.com/Dao-AILab/flash-attention@v2.6.3#subdirectory=csrc/fused_dense_lib
|
22 |
+
jaxtyping
|
23 |
+
mediapipe
|
24 |
+
gradio
|
25 |
+
git+https://github.com/facebookresearch/pytorch3d.git
|
26 |
+
opencv-python==4.5.5.64
|
27 |
+
opencv-python-headless==4.5.5.64
|