Spaces:
Running
Running
David Krajewski
commited on
Commit
•
529e347
1
Parent(s):
d668d8e
changed gpu time
Browse files
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 |
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=
|
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()
|