YulianSa commited on
Commit
65b1e2e
·
1 Parent(s): fdc86f3
Files changed (1) hide show
  1. infer_api.py +1 -1
infer_api.py CHANGED
@@ -554,7 +554,7 @@ def get_distract_mask(generator, color_0, color_1, normal_0=None, normal_1=None,
554
  # )
555
  infer_refine_outside_ratio = 0.20
556
 
557
- @spaces.GPU()
558
  def infer_refine(meshes, imgs):
559
  fixed_v, fixed_f, fixed_t = None, None, None
560
  flow_vert, flow_vector = None, None
 
554
  # )
555
  infer_refine_outside_ratio = 0.20
556
 
557
+ @spaces.GPU(duration=70)
558
  def infer_refine(meshes, imgs):
559
  fixed_v, fixed_f, fixed_t = None, None, None
560
  flow_vert, flow_vector = None, None