lehduong commited on
Commit
0d59d50
1 Parent(s): 1f77d21

Update requirements.txt

Browse files
Files changed (1) hide show
  1. requirements.txt +27 -6
requirements.txt CHANGED
@@ -1,6 +1,27 @@
1
- transformers
2
- diffusers
3
- peft
4
- opencv-python
5
- protobuf
6
- sentencepiece
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
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