Spaces:
Running
on
Zero
Running
on
Zero
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:
|