Spaces:
Running
Running
Update requirements.txt
Browse files- requirements.txt +4 -2
requirements.txt
CHANGED
@@ -1,4 +1,5 @@
|
|
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
|
@@ -17,7 +18,7 @@ peft
|
|
17 |
Pillow
|
18 |
pygltflib
|
19 |
pymeshlab>=2023.12
|
20 |
-
rembg[gpu]
|
21 |
torch_scatter @ https://data.pyg.org/whl/torch-2.2.0%2Bcu121/torch_scatter-2.1.2%2Bpt22cu121-cp310-cp310-linux_x86_64.whl
|
22 |
tqdm
|
23 |
transformers
|
@@ -25,4 +26,5 @@ trimesh
|
|
25 |
typeguard
|
26 |
wandb
|
27 |
xformers
|
28 |
-
ninja
|
|
|
|
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
|
|
|
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
|
|
|
26 |
typeguard
|
27 |
wandb
|
28 |
xformers
|
29 |
+
ninja
|
30 |
+
spaces # ZeroGPU를 위해 추가
|