1inkusFace commited on
Commit
30904d8
·
verified ·
1 Parent(s): 6b44565

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +5 -5
app.py CHANGED
@@ -15,9 +15,9 @@ from diffusers.utils import load_image
15
  #task_type = None
16
 
17
  @spaces.GPU(duration=120)
18
- #def init_predictor():
19
- # global predictor
20
- predictor = SkyReelsVideoSingleGpuInfer(
21
  task_type= TaskType.I2V,
22
  model_id="Skywork/SkyReels-V1-Hunyuan-I2V",
23
  quant_model=True,
@@ -27,7 +27,7 @@ predictor = SkyReelsVideoSingleGpuInfer(
27
  parameters_level=True,
28
  compiler_transformer=False,
29
  )
30
- )
31
 
32
  @spaces.GPU(duration=80)
33
  def generate_video(prompt, seed, image=None):
@@ -77,6 +77,6 @@ def create_gradio_interface():
77
  if __name__ == "__main__":
78
  #import multiprocessing
79
  #multiprocessing.freeze_support()
80
- #init_predictor()
81
  demo = create_gradio_interface()
82
  demo.launch()
 
15
  #task_type = None
16
 
17
  @spaces.GPU(duration=120)
18
+ def init_predictor():
19
+ global predictor
20
+ predictor = SkyReelsVideoSingleGpuInfer(
21
  task_type= TaskType.I2V,
22
  model_id="Skywork/SkyReels-V1-Hunyuan-I2V",
23
  quant_model=True,
 
27
  parameters_level=True,
28
  compiler_transformer=False,
29
  )
30
+ )
31
 
32
  @spaces.GPU(duration=80)
33
  def generate_video(prompt, seed, image=None):
 
77
  if __name__ == "__main__":
78
  #import multiprocessing
79
  #multiprocessing.freeze_support()
80
+ init_predictor()
81
  demo = create_gradio_interface()
82
  demo.launch()