Spaces:
Running
on
Zero
Running
on
Zero
update
Browse files- app.py +10 -0
- requirements.txt +1 -1
app.py
CHANGED
@@ -25,6 +25,16 @@ os.system('pip install ./submodules/diff-surfel-rasterization')
|
|
25 |
# import simple_knn
|
26 |
# import diff_surfel_rasterization
|
27 |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
28 |
import argparse
|
29 |
import spaces
|
30 |
import json
|
|
|
25 |
# import simple_knn
|
26 |
# import diff_surfel_rasterization
|
27 |
|
28 |
+
import torch
|
29 |
+
pyt_version_str=torch.__version__.split("+")[0].replace(".", "")
|
30 |
+
version_str="".join([
|
31 |
+
f"py3{sys.version_info.minor}_cu",
|
32 |
+
torch.version.cuda.replace(".",""),
|
33 |
+
f"_pyt{pyt_version_str}"
|
34 |
+
])
|
35 |
+
os.system('pip install iopath')
|
36 |
+
os.system(f'pip install --no-index --no-cache-dir pytorch3d -f https://dl.fbaipublicfiles.com/pytorch3d/packaging/wheels/{version_str}/download.html')
|
37 |
+
|
38 |
import argparse
|
39 |
import spaces
|
40 |
import json
|
requirements.txt
CHANGED
@@ -45,4 +45,4 @@ tritonclient
|
|
45 |
gradio==4.43.0
|
46 |
rembg
|
47 |
onnxruntime
|
48 |
-
git+https://github.com/facebookresearch/pytorch3d.git@stable
|
|
|
45 |
gradio==4.43.0
|
46 |
rembg
|
47 |
onnxruntime
|
48 |
+
# git+https://github.com/facebookresearch/pytorch3d.git@stable
|