Spaces:
Running
on
Zero
Running
on
Zero
Update models/pipelines.py
Browse files- models/pipelines.py +0 -2
models/pipelines.py
CHANGED
|
@@ -24,7 +24,6 @@ from models.cogvideox_tracking import CogVideoXImageToVideoPipelineTracking
|
|
| 24 |
from submodules.MoGe.moge.model import MoGeModel
|
| 25 |
from image_gen_aux import DepthPreprocessor
|
| 26 |
from moviepy.editor import ImageSequenceClip
|
| 27 |
-
import space
|
| 28 |
|
| 29 |
class DiffusionAsShaderPipeline:
|
| 30 |
def __init__(self, gpu_id=0, output_dir='outputs'):
|
|
@@ -218,7 +217,6 @@ class DiffusionAsShaderPipeline:
|
|
| 218 |
return tracking_pts
|
| 219 |
|
| 220 |
##============= SpatialTracker =============##
|
| 221 |
-
@spaces.GPU
|
| 222 |
def generate_tracking_spatracker(self, video_tensor, density=70):
|
| 223 |
"""Generate tracking video
|
| 224 |
|
|
|
|
| 24 |
from submodules.MoGe.moge.model import MoGeModel
|
| 25 |
from image_gen_aux import DepthPreprocessor
|
| 26 |
from moviepy.editor import ImageSequenceClip
|
|
|
|
| 27 |
|
| 28 |
class DiffusionAsShaderPipeline:
|
| 29 |
def __init__(self, gpu_id=0, output_dir='outputs'):
|
|
|
|
| 217 |
return tracking_pts
|
| 218 |
|
| 219 |
##============= SpatialTracker =============##
|
|
|
|
| 220 |
def generate_tracking_spatracker(self, video_tensor, density=70):
|
| 221 |
"""Generate tracking video
|
| 222 |
|