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,
|