1inkusFace commited on
Commit
33acb21
·
verified ·
1 Parent(s): 3587696

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -7,7 +7,7 @@ import os
7
  import random
8
  from skyreelsinfer.offload import OffloadConfig
9
  from skyreelsinfer import TaskType
10
- from skyreelsinfer.skyreels_video_infer import SkyReelsVideoInfer
11
  from diffusers.utils import export_to_video
12
  from diffusers.utils import load_image
13
 
@@ -17,7 +17,7 @@ task_type = None
17
  @spaces.GPU()
18
  def init_predictor():
19
  global predictor
20
- predictor = SkyReelsVideoInfer(
21
  task_type= TaskType.I2V,
22
  model_id="Skywork/SkyReels-V1-Hunyuan-I2V",
23
  quant_model=True,
 
7
  import random
8
  from skyreelsinfer.offload import OffloadConfig
9
  from skyreelsinfer import TaskType
10
+ from skyreelsinfer.skyreels_video_infer import SkyReelsVideoSingleGpuInfer
11
  from diffusers.utils import export_to_video
12
  from diffusers.utils import load_image
13
 
 
17
  @spaces.GPU()
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,