Spaces:
Paused
Paused
Update app.py
Browse files
app.py
CHANGED
@@ -15,9 +15,9 @@ from diffusers.utils import load_image
|
|
15 |
#task_type = None
|
16 |
|
17 |
@spaces.GPU(duration=120)
|
18 |
-
|
19 |
-
|
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 |
-
|
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()
|