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

Update requirements.txt

Browse files
Files changed (1) hide show
  1. requirements.txt +7 -6
requirements.txt CHANGED
@@ -1,8 +1,9 @@
1
  pytorch3d @ https://dl.fbaipublicfiles.com/pytorch3d/packaging/wheels/py310_cu121_pyt221/pytorch3d-0.7.6-cp310-cp310-linux_x86_64.whl
2
- # GPU 버전들은 유지 (ZeroGPU가 런타임에 할당)
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 @ 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
5
  torch==2.2.0
 
6
  accelerate
7
  datasets
8
  diffusers>=0.26.3
@@ -18,8 +19,7 @@ peft
18
  Pillow
19
  pygltflib
20
  pymeshlab>=2023.12
21
- rembg # [gpu] 제거!
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
@@ -27,4 +27,5 @@ typeguard
27
  wandb
28
  xformers
29
  ninja
30
- spaces # ZeroGPU를 위해 추가
 
 
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
 
19
  Pillow
20
  pygltflib
21
  pymeshlab>=2023.12
22
+ rembg
 
23
  tqdm
24
  transformers
25
  trimesh
 
27
  wandb
28
  xformers
29
  ninja
30
+ spaces
31
+ gradio_client