Spaces:
Running
on
Zero
Running
on
Zero
gc
Browse files
app.py
CHANGED
|
@@ -18,6 +18,8 @@ import wan
|
|
| 18 |
from wan.configs import WAN_CONFIGS, SIZE_CONFIGS, MAX_AREA_CONFIGS, SUPPORTED_SIZES
|
| 19 |
from wan.utils.utils import cache_video
|
| 20 |
|
|
|
|
|
|
|
| 21 |
# --- 1. Global Setup and Model Loading ---
|
| 22 |
|
| 23 |
print("Starting Gradio App for Wan 2.2 TI2V-5B...")
|
|
@@ -154,7 +156,8 @@ def generate_video(
|
|
| 154 |
normalize=True,
|
| 155 |
value_range=(-1, 1)
|
| 156 |
)
|
| 157 |
-
|
|
|
|
| 158 |
return video_path
|
| 159 |
|
| 160 |
|
|
|
|
| 18 |
from wan.configs import WAN_CONFIGS, SIZE_CONFIGS, MAX_AREA_CONFIGS, SUPPORTED_SIZES
|
| 19 |
from wan.utils.utils import cache_video
|
| 20 |
|
| 21 |
+
import gc
|
| 22 |
+
|
| 23 |
# --- 1. Global Setup and Model Loading ---
|
| 24 |
|
| 25 |
print("Starting Gradio App for Wan 2.2 TI2V-5B...")
|
|
|
|
| 156 |
normalize=True,
|
| 157 |
value_range=(-1, 1)
|
| 158 |
)
|
| 159 |
+
del video_tensor
|
| 160 |
+
gc.collect()
|
| 161 |
return video_path
|
| 162 |
|
| 163 |
|