init project
Browse files- modules/pe3r/demo.py +1 -1
- modules/pe3r/models.py +1 -1
modules/pe3r/demo.py
CHANGED
@@ -549,7 +549,7 @@ def set_scenegraph_options(inputfiles, winsize, refid, scenegraph_type):
|
|
549 |
return winsize, refid
|
550 |
|
551 |
|
552 |
-
@spaces.GPU(duration=
|
553 |
def main_demo(tmpdirname, pe3r, server_name, server_port, silent=False):
|
554 |
device = 'cuda' if torch.cuda.is_available() else 'cpu'
|
555 |
# scene, outfile, imgs = get_reconstructed_scene(
|
|
|
549 |
return winsize, refid
|
550 |
|
551 |
|
552 |
+
@spaces.GPU(duration=180)
|
553 |
def main_demo(tmpdirname, pe3r, server_name, server_port, silent=False):
|
554 |
device = 'cuda' if torch.cuda.is_available() else 'cpu'
|
555 |
# scene, outfile, imgs = get_reconstructed_scene(
|
modules/pe3r/models.py
CHANGED
@@ -14,7 +14,7 @@ import spaces
|
|
14 |
import torch
|
15 |
|
16 |
class Models:
|
17 |
-
@spaces.GPU(duration=
|
18 |
def __init__(self, device='cpu'):
|
19 |
device = 'cuda' if torch.cuda.is_available() else 'cpu'
|
20 |
# -- mast3r --
|
|
|
14 |
import torch
|
15 |
|
16 |
class Models:
|
17 |
+
@spaces.GPU(duration=180)
|
18 |
def __init__(self, device='cpu'):
|
19 |
device = 'cuda' if torch.cuda.is_available() else 'cpu'
|
20 |
# -- mast3r --
|