yslan commited on
Commit
bbd5eb0
1 Parent(s): a0896dd
Files changed (2) hide show
  1. app.py +10 -0
  2. 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