EXCAI commited on
Commit
b23950a
·
verified ·
1 Parent(s): 0932790

Update models/pipelines.py

Browse files
Files changed (1) hide show
  1. models/pipelines.py +1 -0
models/pipelines.py CHANGED
@@ -24,6 +24,7 @@ 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
 
28
  class DiffusionAsShaderPipeline:
29
  def __init__(self, gpu_id=0, output_dir='outputs'):
 
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'):