David Krajewski commited on
Commit
529e347
1 Parent(s): d668d8e

changed gpu time

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -236,7 +236,7 @@ class Drag:
236
  self.width = width
237
  self.model_length = model_length
238
 
239
- @spaces.GPU(enable_queue=True, duration=240)
240
  def get_cmp_flow(self, frames, sparse_optical_flow, mask, brush_mask=None):
241
 
242
  '''
@@ -507,7 +507,7 @@ class Drag:
507
 
508
  print("models initialized!")
509
 
510
- @spaces.GPU(enable_queue=True, duration=240)
511
  def run(self, first_frame_path, tracking_points, inference_batch_size, motion_brush_mask, motion_brush_viz, ctrl_scale):
512
  if not self.cmp or not self.pipeline:
513
  self.run_model_init()
 
236
  self.width = width
237
  self.model_length = model_length
238
 
239
+ @spaces.GPU(enable_queue=True, duration=120)
240
  def get_cmp_flow(self, frames, sparse_optical_flow, mask, brush_mask=None):
241
 
242
  '''
 
507
 
508
  print("models initialized!")
509
 
510
+ @spaces.GPU(enable_queue=True, duration=120)
511
  def run(self, first_frame_path, tracking_points, inference_batch_size, motion_brush_mask, motion_brush_viz, ctrl_scale):
512
  if not self.cmp or not self.pipeline:
513
  self.run_model_init()