fffiloni commited on
Commit
9afd27b
·
verified ·
1 Parent(s): c9d89c3

pre-download 1.3B model on shared ui

Browse files
Files changed (1) hide show
  1. app_df.py +4 -0
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,