Drexubery commited on
Commit
b20f9b9
1 Parent(s): ff61960
Files changed (1) hide show
  1. app.py +13 -4
app.py CHANGED
@@ -1,13 +1,22 @@
1
  import os
2
- os.system('conda install https://anaconda.org/pytorch3d/pytorch3d/0.7.7/download/linux-64/pytorch3d-0.7.7-py39_cu117_pyt200.tar.bz2')
3
- os.system('wget https://download.europe.naverlabs.com/ComputerVision/DUSt3R/DUSt3R_ViTLarge_BaseDecoder_512_dpt.pth -P checkpoints/')
4
-
5
  import sys
 
 
 
 
 
 
 
 
 
 
 
 
6
  import gradio as gr
7
  import random
8
  from viewcrafter import ViewCrafter
9
  from configs.infer_config import get_parser
10
- import torch
11
  from huggingface_hub import hf_hub_download
12
 
13
  i2v_examples = [
 
1
  import os
2
+ import torch
 
 
3
  import sys
4
+
5
+ pyt_version_str=torch.__version__.split("+")[0].replace(".", "")
6
+ version_str="".join([
7
+ f"py3{sys.version_info.minor}_cu",
8
+ torch.version.cuda.replace(".",""),
9
+ f"_pyt{pyt_version_str}"
10
+ ])
11
+
12
+ os.system('pip install iopath')
13
+ os.system(f'pip install --no-index --no-cache-dir pytorch3d -f https://dl.fbaipublicfiles.com/pytorch3d/packaging/wheels/{version_str}/download.html')
14
+ # os.system('wget https://download.europe.naverlabs.com/ComputerVision/DUSt3R/DUSt3R_ViTLarge_BaseDecoder_512_dpt.pth -P checkpoints/')
15
+
16
  import gradio as gr
17
  import random
18
  from viewcrafter import ViewCrafter
19
  from configs.infer_config import get_parser
 
20
  from huggingface_hub import hf_hub_download
21
 
22
  i2v_examples = [