Daniellesry commited on
Commit
3433f72
·
1 Parent(s): 3c1a3b0

update moge code

Browse files
Files changed (1) hide show
  1. dkt/pipelines/pipeline.py +2 -1
dkt/pipelines/pipeline.py CHANGED
@@ -909,7 +909,7 @@ class DKTPipeline:
909
 
910
 
911
 
912
- @spaces.GPU(duration=240)
913
  @torch.inference_mode()
914
  def __call__(self, video_file,\
915
  negative_prompt='', height=480, width=832, \
@@ -957,6 +957,7 @@ class DKTPipeline:
957
  sliding_window_stride=window_size - overlap,
958
  cfg_scale=1.0,
959
  )
 
960
  print(f"[P4] main_pipe done, len(video)={len(video)}")
961
  torch.cuda.empty_cache()
962
 
 
909
 
910
 
911
 
912
+ @spaces.GPU(duration=180)
913
  @torch.inference_mode()
914
  def __call__(self, video_file,\
915
  negative_prompt='', height=480, width=832, \
 
957
  sliding_window_stride=window_size - overlap,
958
  cfg_scale=1.0,
959
  )
960
+ del vae_outs
961
  print(f"[P4] main_pipe done, len(video)={len(video)}")
962
  torch.cuda.empty_cache()
963