Spaces:
Running
Running
Update requirements.txt
Browse files- 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 |
-
#
|
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 |
-
|
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
|
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
|
|
|
|
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
|