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