Update app.py
Browse files
app.py
CHANGED
|
@@ -119,6 +119,9 @@ def load_t2v_pipeline0():
|
|
| 119 |
import time
|
| 120 |
import torch
|
| 121 |
|
|
|
|
|
|
|
|
|
|
| 122 |
def load_t2v_pipeline():
|
| 123 |
"""Always load the T2V pipeline at startup and store in cache."""
|
| 124 |
start_time = time.time()
|
|
@@ -177,6 +180,8 @@ def load_t2v_pipeline():
|
|
| 177 |
|
| 178 |
clear_memory()
|
| 179 |
|
|
|
|
|
|
|
| 180 |
def load_i2v_pipeline():
|
| 181 |
"""Load and optimize the I2V pipeline."""
|
| 182 |
if i2v_pipe_cache[0] is None:
|
|
|
|
| 119 |
import time
|
| 120 |
import torch
|
| 121 |
|
| 122 |
+
import time
|
| 123 |
+
import torch
|
| 124 |
+
|
| 125 |
def load_t2v_pipeline():
|
| 126 |
"""Always load the T2V pipeline at startup and store in cache."""
|
| 127 |
start_time = time.time()
|
|
|
|
| 180 |
|
| 181 |
clear_memory()
|
| 182 |
|
| 183 |
+
return t2v_pipe_cache[0] # ✅ return the cached pipeline
|
| 184 |
+
|
| 185 |
def load_i2v_pipeline():
|
| 186 |
"""Load and optimize the I2V pipeline."""
|
| 187 |
if i2v_pipe_cache[0] is None:
|