From b562190b93021b47cb77fd4a51f49ffe3492e143 Mon Sep 17 00:00:00 2001
From: DeepBeepMeep <deepbeepmeep@yahoo.com>
Date: Wed, 18 Jun 2025 00:20:53 +0200
Subject: [PATCH] Fixed skyreels gen bug

---
 wan/diffusion_forcing.py | 2 +-
 wgp.py                   | 2 +-
 2 files changed, 2 insertions(+), 2 deletions(-)

diff --git a/wan/diffusion_forcing.py b/wan/diffusion_forcing.py
index 3c6f701..774a444 100644
--- a/wan/diffusion_forcing.py
+++ b/wan/diffusion_forcing.py
@@ -236,7 +236,7 @@ class DTT2V:
         if input_video != None:
             _ , _ , height, width  = input_video.shape
         elif image_start != None:
-            image_start = image_start[0]
+            image_start = image_start
             frame_width, frame_height  = image_start.size
             height, width = calculate_new_dimensions(height, width, frame_height, frame_width, fit_into_canvas)
             image_start = np.array(image_start.resize((width, height))).transpose(2, 0, 1)
diff --git a/wgp.py b/wgp.py
index 49893c9..9903c5c 100644
--- a/wgp.py
+++ b/wgp.py
@@ -302,7 +302,7 @@ def process_prompt_and_add_tasks(state, model_choice):
     if test_any_sliding_window(model_type):
         if video_length > sliding_window_size:
             no_windows = compute_sliding_window_no(video_length, sliding_window_size, sliding_window_discard_last_frames, sliding_window_overlap)
-            gr.Info(f"The Number of Frames to generate ({video_length}) is greater than the Sliding Window Size ({sliding_window_size}) , {no_windows} Windows will be generated")
+            gr.Info(f"The Number of Frames to generate ({video_length}) is greater than the Sliding Window Size ({sliding_window_size}), {no_windows} Windows will be generated")
 
     if "recam" in model_filename:
         if video_source == None: