Spaces:
Running
on
Zero
Running
on
Zero
update
Browse files
app.py
CHANGED
@@ -1,13 +1,22 @@
|
|
1 |
import os
|
2 |
-
|
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 = [
|