Painimy commited on
Commit
10c717b
·
verified ·
1 Parent(s): fc219b8

Update requirements.txt

Browse files
Files changed (1) hide show
  1. requirements.txt +8 -10
requirements.txt CHANGED
@@ -1,29 +1,27 @@
1
- pytorch3d @ https://dl.fbaipublicfiles.com/pytorch3d/packaging/wheels/py310_cu121_pyt221/pytorch3d-0.7.6-cp310-cp310-linux_x86_64.whl
2
- ort_nightly_gpu @ https://aiinfra.pkgs.visualstudio.com/2692857e-05ef-43b4-ba9c-ccf1c22c437c/_packaging/d3daa2b0-aa56-45ac-8145-2c3dc0661c87/pypi/download/ort-nightly-gpu/1.17.dev20240118002/ort_nightly_gpu-1.17.0.dev20240118002-cp310-cp310-manylinux_2_28_x86_64.whl
3
- onnxruntime_gpu @ https://pkgs.dev.azure.com/onnxruntime/2a773b67-e88b-4c7f-9fc0-87d31fea8ef2/_packaging/7fa31e42-5da1-4e84-a664-f2b4129c7d45/pypi/download/onnxruntime-gpu/1.17/onnxruntime_gpu-1.17.0-cp310-cp310-manylinux_2_28_x86_64.whl
4
- torch==2.2.0
5
  accelerate
6
  datasets
7
  diffusers>=0.26.3
8
  fire
9
- gradio
10
  jaxtyping
11
  numba
12
  numpy
13
  omegaconf>=2.3.0
14
- opencv_python
15
- opencv_python_headless
16
  peft
17
  Pillow
18
  pygltflib
19
  pymeshlab>=2023.12
20
  rembg[gpu]
21
- torch>=2.0.1
22
- torch_scatter @ https://data.pyg.org/whl/torch-2.2.0%2Bcu121/torch_scatter-2.1.2%2Bpt22cu121-cp310-cp310-linux_x86_64.whl
23
  tqdm
24
  transformers
25
  trimesh
26
  typeguard
27
  wandb
28
  xformers
29
- ninja
 
1
+ gradio>=4.24.0
2
+ torch>=2.0.0
3
+ pytorch3d
4
+ onnxruntime-gpu
5
  accelerate
6
  datasets
7
  diffusers>=0.26.3
8
  fire
 
9
  jaxtyping
10
  numba
11
  numpy
12
  omegaconf>=2.3.0
13
+ opencv-python
14
+ opencv-python-headless
15
  peft
16
  Pillow
17
  pygltflib
18
  pymeshlab>=2023.12
19
  rembg[gpu]
20
+ torch-scatter
 
21
  tqdm
22
  transformers
23
  trimesh
24
  typeguard
25
  wandb
26
  xformers
27
+ ninja