Spaces:
Running
Running
pre-download 1.3B model on shared ui
Browse files
app_df.py
CHANGED
@@ -14,6 +14,10 @@ from skyreels_v2_infer.pipelines import PromptEnhancer, resizecrop
|
|
14 |
is_shared_ui = True if "fffiloni/SkyReels-V2" in os.environ['SPACE_ID'] else False
|
15 |
#is_shared_ui = False
|
16 |
|
|
|
|
|
|
|
|
|
17 |
def generate_diffusion_forced_video(
|
18 |
prompt,
|
19 |
model_id,
|
|
|
14 |
is_shared_ui = True if "fffiloni/SkyReels-V2" in os.environ['SPACE_ID'] else False
|
15 |
#is_shared_ui = False
|
16 |
|
17 |
+
model_id = None
|
18 |
+
if is_shared_ui:
|
19 |
+
model_id = download_model("Skywork/SkyReels-V2-DF-1.3B-540P")
|
20 |
+
|
21 |
def generate_diffusion_forced_video(
|
22 |
prompt,
|
23 |
model_id,
|