Update app_df.py

app_df.py

@@ -12,7 +12,7 @@ from skyreels_v2_infer.modules import download_model
 from skyreels_v2_infer.pipelines import PromptEnhancer, resizecrop
 
 is_shared_ui = True if "fffiloni/SkyReels-V2" in os.environ['SPACE_ID'] else False
-is_shared_ui = False
+#is_shared_ui = False
 
 def generate_diffusion_forced_video(
     prompt,
@@ -195,9 +195,9 @@ with gr.Blocks() as demo:
                     fps = gr.Slider(minimum=1, maximum=60, value=24, step=1, label="FPS")
                     seed = gr.Number(label="Seed (optional)", precision=0)
                     prompt_enhancer = gr.Checkbox(label="Prompt Enhancer")
-                    use_teacache = gr.Checkbox(label="Use TeaCache")
-                    teacache_thresh = gr.Slider(minimum=0.0, maximum=1.0, value=0.3, step=0.01, label="TeaCache Threshold")
-                    use_ret_steps = gr.Checkbox(label="Use Retention Steps")
+                    use_teacache = gr.Checkbox(label="Use TeaCache", value=True)
+                    teacache_thresh = gr.Slider(minimum=0.0, maximum=1.0, value=0.2, step=0.01, label="TeaCache Threshold")
+                    use_ret_steps = gr.Checkbox(label="Use Retention Steps", value=True)
 
                 submit_btn = gr.Button("Generate")
 
@@ -221,13 +221,17 @@ with gr.Blocks() as demo:
         overlap_history = 0
         addnoise_condition = 0
         ar_step = 0
+        causal_attention = False
         causal_block_size = 1
-        use_teacache = False
-        teacache_thresh = 0.3
-        use_ret_steps = False
+        use_teacache = True
+        teacache_thresh = 0.2
+        use_ret_steps = True
 
         if target_l == "4":
             n_frames = 97
+            use_teacache = True
+            teacache_thresh = 0.2
+            use_ret_steps = True
         elif target_l == "10":
             n_frames = 257
             overlap_history = 17
@@ -249,6 +253,7 @@ with gr.Blocks() as demo:
             use_teacache = True
             teacache_thresh = 0.3
             use_ret_steps = True
+            causal_attention = True
             ar_step = 5
             causal_block_size = 5
         elif target_l == "60":
@@ -258,16 +263,17 @@ with gr.Blocks() as demo:
             use_teacache = True
             teacache_thresh = 0.3
             use_ret_steps = True
+            causal_attention = True
             ar_step = 5
             causal_block_size = 5
         
-        return n_frames, overlap_history, addnoise_condition, ar_step, causal_block_size, use_teacache, teacache_thresh, use_ret_steps
+        return n_frames, overlap_history, addnoise_condition, ar_step, causal_attention, causal_block_size, use_teacache, teacache_thresh, use_ret_steps
         
 
     target_length.change(
         fn = set_num_frames,
         inputs = [target_length],
-        outputs = [num_frames, overlap_history, addnoise_condition, ar_step, causal_block_size, use_teacache, teacache_thresh, use_ret_steps],
+        outputs = [num_frames, overlap_history, addnoise_condition, ar_step, causal_attention, causal_block_size, use_teacache, teacache_thresh, use_ret_steps],
         queue = False
     )