RageshAntony commited on
Commit
4adb0b3
·
verified ·
1 Parent(s): 3fad153

added pytoch3d

Browse files
Files changed (1) hide show
  1. requirements.txt +3 -3
requirements.txt CHANGED
@@ -5,8 +5,8 @@ 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,6 +16,6 @@ scikit-learn
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
 
 
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
  xformers
9
+ numpy
10
  opencv-python
11
  peft
12
  sentencepiece
 
16
  scipy
17
  einops
18
  jaxtyping
 
19
  bitsandbytes
20
+ beautifulsoup4
21
+ git+https://github.com/facebookresearch/pytorch3d.git@stable