lehduong commited on
Commit
619d312
1 Parent(s): e35479d

Update requirements.txt

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