Spaces:
Sleeping
Sleeping
Commit
·
3433f72
1
Parent(s):
3c1a3b0
update moge code
Browse files
dkt/pipelines/pipeline.py
CHANGED
|
@@ -909,7 +909,7 @@ class DKTPipeline:
|
|
| 909 |
|
| 910 |
|
| 911 |
|
| 912 |
-
@spaces.GPU(duration=
|
| 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 |
|