add gpu for spatracker

app.py

@@ -31,15 +31,15 @@ GPU_ID = args.gpu
 DEFAULT_MODEL_PATH = args.model_path
 OUTPUT_DIR = args.output_dir
 
-if 'CUDA_HOME' not in os.environ:
-    for cuda_path in ['/usr/local/cuda', '/usr/cuda', '/opt/cuda']:
-        if os.path.exists(cuda_path):
-            os.environ['CUDA_HOME'] = cuda_path
-            print(cuda_path)
-            break
-    if 'CUDA_HOME' not in os.environ:
-        os.environ['CUDA_HOME'] = '/usr/local/cuda'
-        print("set default cuda path in: /usr/local/cuda")
+# if 'CUDA_HOME' not in os.environ:
+#     for cuda_path in ['/usr/local/cuda', '/usr/cuda', '/opt/cuda']:
+#         if os.path.exists(cuda_path):
+#             os.environ['CUDA_HOME'] = cuda_path
+#             print(cuda_path)
+#             break
+#     if 'CUDA_HOME' not in os.environ:
+#         os.environ['CUDA_HOME'] = '/usr/local/cuda'
+#         print("set default cuda path in: /usr/local/cuda")
 
 # Create necessary directories
 os.makedirs("outputs", exist_ok=True)
@@ -110,7 +110,7 @@ def create_run_command(args):
     
     return cmd
 
-@spaces.GPU(duration=280)
+@spaces.GPU(duration=270)
 def run_process(cmd):
     """Run command and return output"""
     print(f"Running command: {' '.join(cmd)}")
@@ -138,7 +138,7 @@ def run_process(cmd):
     
     return "\n".join(output)
 
-@spaces.GPU(duration=280)
+@spaces.GPU(duration=270)
 def process_motion_transfer(source, prompt, mt_repaint_option, mt_repaint_image):
     """Process video motion transfer task"""
     try:
@@ -185,7 +185,7 @@ def process_motion_transfer(source, prompt, mt_repaint_option, mt_repaint_image)
         print(f"Processing failed: {str(e)}\n{traceback.format_exc()}")
         return None
 
-@spaces.GPU(duration=280)
+@spaces.GPU(duration=270)
 def process_camera_control(source, prompt, camera_motion, tracking_method):
     """Process camera control task"""
     try:
@@ -226,7 +226,7 @@ def process_camera_control(source, prompt, camera_motion, tracking_method):
         print(f"Processing failed: {str(e)}\n{traceback.format_exc()}")
         return None
 
-@spaces.GPU(duration=280)
+@spaces.GPU(duration=270)
 def process_object_manipulation(source, prompt, object_motion, object_mask, tracking_method):
     """Process object manipulation task"""
     try:
@@ -265,7 +265,7 @@ def process_object_manipulation(source, prompt, object_motion, object_mask, trac
         print(f"Processing failed: {str(e)}\n{traceback.format_exc()}")
         return None
 
-@spaces.GPU(duration=280)
+@spaces.GPU(duration=270)
 def process_mesh_animation(source, prompt, tracking_video, ma_repaint_option, ma_repaint_image):
     """Process mesh animation task"""
     try:

models/pipelines.py

@@ -217,7 +217,7 @@ class DiffusionAsShaderPipeline:
         return tracking_pts
     
     ##============= SpatialTracker =============##
-    
+    @spaces.GPU
     def generate_tracking_spatracker(self, video_tensor, density=70):
         """Generate tracking video