hujiecpp commited on
Commit
b65ac4b
·
1 Parent(s): 722c089

init project

Browse files
Files changed (2) hide show
  1. modules/pe3r/demo.py +1 -1
  2. 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=300)
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=300)
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 --