Spaces:
Paused
Paused
Update app.py
Browse files
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
|
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 =
|
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,
|