Update models/pipelines.py

models/pipelines.py

@@ -24,7 +24,6 @@ from models.cogvideox_tracking import CogVideoXImageToVideoPipelineTracking
 from submodules.MoGe.moge.model import MoGeModel
 from image_gen_aux import DepthPreprocessor
 from moviepy.editor import ImageSequenceClip
-import space
 
 class DiffusionAsShaderPipeline:
     def __init__(self, gpu_id=0, output_dir='outputs'):
@@ -218,7 +217,6 @@ class DiffusionAsShaderPipeline:
         return tracking_pts
     
     ##============= SpatialTracker =============##
-    @spaces.GPU
     def generate_tracking_spatracker(self, video_tensor, density=70):
         """Generate tracking video