small fix

app.py

@@ -394,7 +394,7 @@ def video_generation_handler(prompt, seed=42, fps=15, width=DEFAULT_WIDTH, heigh
 
 # --- Gradio UI Layout ---
 with gr.Blocks(title="Wan2.1 1.3B Self-Forcing demo") as demo:
-    gr.Markdown("Video generation with distilled Wan2-1 1.3B [[Model]](https://huggingface.co/gdhe17/Self-Forcing), [[Project page]](https://self-forcing.github.io), [[Paper]](https://huggingface.co/papers/2506.08009))")
+    gr.Markdown("Video generation with distilled Wan2-1 1.3B [[Model]](https://huggingface.co/gdhe17/Self-Forcing), [[Project page]](https://self-forcing.github.io), [[Paper]](https://huggingface.co/papers/2506.08009)")
     
     with gr.Row():
         with gr.Column(scale=2):
@@ -409,11 +409,12 @@ with gr.Blocks(title="Wan2.1 1.3B Self-Forcing demo") as demo:
             
             gr.Markdown("### ⚙️ Settings")
             with gr.Row():
-                seed = gr.Number(
-                    label="Seed", 
-                    value=-1, 
-                    info="Use -1 for random seed",
-                    precision=0
+                seed = gr.Slider(
+                    label="Generation Seed (-1 for random)",
+                    minimum=-1,
+                    maximum=2147483647,  # 2^31 - 1
+                    step=1,
+                    value=-1
                 )
                 fps = gr.Slider(
                     label="Playback FPS", 

app_with_streaming.py

@@ -263,7 +263,11 @@ def video_generation_handler_streaming(prompt, seed=42, fps=15, width=DEFAULT_WI
     rnd = torch.Generator(gpu).manual_seed(int(seed))
     pipeline._initialize_kv_cache(1, torch.float16, device=gpu)
     pipeline._initialize_crossattn_cache(1, torch.float16, device=gpu)
-    noise = torch.randn([1, 21, 16, 60, 104], device=gpu, dtype=torch.float16, generator=rnd)
+    
+    # Calculate latent dimensions based on actual width/height (assuming 8x downsampling)
+    latent_height = height // 8
+    latent_width = width // 8
+    noise = torch.randn([1, 21, 16, latent_height, latent_width], device=gpu, dtype=torch.float16, generator=rnd)
     
     vae_cache, latents_cache = None, None
     if not APP_STATE["current_use_taehv"] and not args.trt:
@@ -432,11 +436,12 @@ with gr.Blocks(title="Wan2.1 1.3B Self-Forcing streaming demo") as demo:
             
             gr.Markdown("### ⚙️ Settings")
             with gr.Row():
-                seed = gr.Number(
-                    label="Seed", 
-                    value=-1, 
-                    info="Use -1 for random seed",
-                    precision=0
+                seed = gr.Slider(
+                    label="Generation Seed (-1 for random)",
+                    minimum=-1,
+                    maximum=2147483647,  # 2^31 - 1
+                    step=1,
+                    value=-1
                 )
                 fps = gr.Slider(
                     label="Playback FPS",