ginipick commited on
Commit
81ec7b3
·
verified ·
1 Parent(s): 49eeb76

Update requirements.txt

Browse files
Files changed (1) hide show
  1. requirements.txt +11 -9
requirements.txt CHANGED
@@ -1,10 +1,8 @@
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와 onnxruntime-gpu 중 하나만 사용 (중복 제거)
3
- # 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
4
  onnxruntime-gpu==1.17.0
5
  torch==2.2.0
6
  torch_scatter @ https://data.pyg.org/whl/torch-2.2.0%2Bcu121/torch_scatter-2.1.2%2Bpt22cu121-cp310-cp310-linux_x86_64.whl
7
- accelerate
8
  datasets
9
  diffusers>=0.26.3
10
  fire
@@ -13,19 +11,23 @@ jaxtyping
13
  numba
14
  numpy<2
15
  omegaconf>=2.3.0
16
- opencv_python
17
- opencv_python_headless
18
- peft
19
  Pillow
20
  pygltflib
21
  pymeshlab>=2023.12
22
  rembg
23
  tqdm
24
- transformers
25
  trimesh
26
  typeguard
27
  wandb
28
- xformers
29
  ninja
30
  spaces
31
- gradio_client
 
 
 
 
 
1
  pytorch3d @ https://dl.fbaipublicfiles.com/pytorch3d/packaging/wheels/py310_cu121_pyt221/pytorch3d-0.7.6-cp310-cp310-linux_x86_64.whl
 
 
2
  onnxruntime-gpu==1.17.0
3
  torch==2.2.0
4
  torch_scatter @ https://data.pyg.org/whl/torch-2.2.0%2Bcu121/torch_scatter-2.1.2%2Bpt22cu121-cp310-cp310-linux_x86_64.whl
5
+ accelerate==0.25.0
6
  datasets
7
  diffusers>=0.26.3
8
  fire
 
11
  numba
12
  numpy<2
13
  omegaconf>=2.3.0
14
+ opencv-python
15
+ opencv-python-headless
16
+ peft==0.7.1
17
  Pillow
18
  pygltflib
19
  pymeshlab>=2023.12
20
  rembg
21
  tqdm
22
+ transformers==4.36.2
23
  trimesh
24
  typeguard
25
  wandb
26
+ xformers==0.0.24
27
  ninja
28
  spaces
29
+ gradio_client
30
+ safetensors
31
+ sentencepiece
32
+ tokenizers
33
+ huggingface-hub