imlixinyang commited on
Commit
c7ea8b7
·
1 Parent(s): 4f3c86a
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -380,7 +380,7 @@ if __name__ == "__main__":
380
  response.headers.add('Access-Control-Allow-Methods', 'GET,PUT,POST,DELETE,OPTIONS')
381
  return response
382
 
383
- @GPU
384
  def generate_wrapper(cameras, n_frame, image, text_prompt, image_index, image_height, image_width, video_output_path=None):
385
  """生成函数的包装器,用于并发控制"""
386
  return generation_system.generate(cameras, n_frame, image, text_prompt, image_index, image_height, image_width, video_output_path)
 
380
  response.headers.add('Access-Control-Allow-Methods', 'GET,PUT,POST,DELETE,OPTIONS')
381
  return response
382
 
383
+ @spaces.GPU
384
  def generate_wrapper(cameras, n_frame, image, text_prompt, image_index, image_height, image_width, video_output_path=None):
385
  """生成函数的包装器,用于并发控制"""
386
  return generation_system.generate(cameras, n_frame, image, text_prompt, image_index, image_height, image_width, video_output_path)