Update app.py

app.py

@@ -276,10 +276,12 @@ def generate_video(
             width=downscaled_width,
             height=downscaled_height,
             num_frames=num_frames,
-            num_inference_steps=num_inference_steps,
+            timesteps=[1000, 993, 987, 981, 975, 909, 725, 0.03],
             decode_timestep=0.05,
             decode_noise_scale=0.025,
+            image_cond_noise_scale=0.0,
             guidance_scale=guidance_scale,
+            guidance_rescale=0.7,
             generator=torch.Generator().manual_seed(seed),
             output_type="latent",
         ).frames
@@ -300,13 +302,14 @@ def generate_video(
             width=upscaled_width,
             height=upscaled_height,
             num_frames=num_frames,
-            denoise_strength=0.4,
-            num_inference_steps=10,
+            denoise_strength=0.999,  # Effectively, 4 inference steps out of 5
+            timesteps=[1000, 909, 725, 421, 0],
             latents=upscaled_latents,
-            decode_timestep = 0.05,
+            decode_timestep=0.05,
+            decode_noise_scale=0.025,
+            image_cond_noise_scale=0.0,
             guidance_scale=guidance_scale,
-            decode_noise_scale = 0.025,
-            image_cond_noise_scale=0.025,
+            guidance_rescale=0.7,
             generator=torch.Generator(device="cuda").manual_seed(seed),
             output_type="np",
         ).frames[0]
@@ -330,7 +333,7 @@ def generate_video(
         return None, seed
 
 # Create Gradio interface
-with gr.Blocks(theme=gr.themes.Ocean(font=[gr.themes.GoogleFont("Lexend Deca"), "sans-serif"]), delete_cache=(60, 1800)) as demo:
+with gr.Blocks(theme=gr.themes.Ocean(font=[gr.themes.GoogleFont("Lexend Deca"), "sans-serif"]), delete_cache=(60, 900)) as demo:
     gr.Markdown(
         """
         # Canny Control LTX Video Distilled
@@ -402,7 +405,8 @@ with gr.Blocks(theme=gr.themes.Ocean(font=[gr.themes.GoogleFont("Lexend Deca"),
                     minimum=5,
                     maximum=10,
                     step=1,
-                    value=7
+                    value=7, 
+                    visible=False
                 )
                 
                 with gr.Row():
@@ -411,7 +415,7 @@ with gr.Blocks(theme=gr.themes.Ocean(font=[gr.themes.GoogleFont("Lexend Deca"),
                         minimum=1.0,
                         maximum=5.0,
                         step=0.1,
-                        value=1.0
+                        value=1.0,
                     )
                     
                 with gr.Row():