RageshAntony commited on
Commit
3fad153
·
verified ·
1 Parent(s): c81b0d7
Files changed (1) hide show
  1. requirements.txt +3 -2
requirements.txt CHANGED
@@ -3,9 +3,10 @@ diffusers
3
  torch
4
  torchvision
5
  transformers
6
- git+https://github.com/huggingface/diffusers.git
7
  https://github.com/Dao-AILab/flash-attention/releases/download/v2.5.9.post1/flash_attn-2.5.9.post1+cu118torch1.12cxx11abiFALSE-cp310-cp310-linux_x86_64.whl
8
  numpy
 
9
  opencv-python
10
  peft
11
  sentencepiece
@@ -15,6 +16,6 @@ scikit-learn
15
  scipy
16
  einops
17
  jaxtyping
18
- git+https://github.com/facebookresearch/pytorch3d.git
19
  bitsandbytes
20
  beautifulsoup4
 
3
  torch
4
  torchvision
5
  transformers
6
+ diffusers
7
  https://github.com/Dao-AILab/flash-attention/releases/download/v2.5.9.post1/flash_attn-2.5.9.post1+cu118torch1.12cxx11abiFALSE-cp310-cp310-linux_x86_64.whl
8
  numpy
9
+ xformers
10
  opencv-python
11
  peft
12
  sentencepiece
 
16
  scipy
17
  einops
18
  jaxtyping
19
+ https://dl.fbaipublicfiles.com/pytorch3d/packaging/wheels/py38_cu113_pyt1110/pytorch3d-0.7.2-cp38-cp38-linux_x86_64.whl
20
  bitsandbytes
21
  beautifulsoup4