diff --git a/gradio_server.py b/gradio_server.py
index fe6c2bb..eec6d14 100644
--- a/gradio_server.py
+++ b/gradio_server.py
@@ -1161,7 +1161,7 @@ def create_demo():
                 video_to_continue = gr.Video(label= "Video to continue", visible= use_image2video and False) #######
                 if args.multiple_images:  
                     image_to_continue = gr.Gallery(
-                            label="Images as a starting point for new videos", type ="pil", #file_types= "image", 
+                            label="Images as a starting point for new videos", type ="numpy", #file_types= "image", 
                             columns=[3], rows=[1], object_fit="contain", height="auto", selected_index=0, interactive= True, visible=use_image2video)
                 else:
                     image_to_continue = gr.Image(label= "Image as a starting point for a new video", visible=use_image2video)
diff --git a/wan/modules/attention.py b/wan/modules/attention.py
index eaeb6c5..5587d1c 100644
--- a/wan/modules/attention.py
+++ b/wan/modules/attention.py
@@ -201,7 +201,7 @@ def pay_attention(
         qkv_list = [q, k, v]
         del q, k , v
         x = sdpa_wrapper( qkv_list, lq).unsqueeze(0)
-    elif attn=="flash" and (version is None or version == 3):
+    elif attn=="flash" and version == 3:
         # Note: dropout_p, window_size are not supported in FA3 now.
         x = flash_attn_interface.flash_attn_varlen_func(
             q=q,