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()
|