rahul7star commited on
Commit
4906825
·
verified ·
1 Parent(s): db9e93a

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +5 -0
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: