Update app.py

app.py

@@ -294,7 +294,6 @@ def generate_video(
             video_pipe = WanImageToVideoPipeline.from_pretrained(
                 VIDEO_MODEL_ID,
                 torch_dtype=torch.bfloat16,
-                variant="fp16",
                 use_safetensors=True
             ).to('cuda')
             
@@ -302,7 +301,9 @@ def generate_video(
             try:
                 video_pipe.load_lora_weights("Kijai/WanVideo_comfy", weight_name="Wan22-Lightning-4-cfg1_bf16_v0.9.safetensors")
                 video_pipe.fuse_lora(lora_scale=1.0)
-            except:
+                print("Lightning LoRA loaded")
+            except Exception as e:
+                print(f"LoRA loading skipped: {e}")
                 pass
         
         # Clear cache before generation
@@ -672,7 +673,6 @@ with gr.Blocks(css=css, theme=gr.themes.Base()) as demo:
     )
     
 
-
 # Launch
 if __name__ == "__main__":
     # Don't initialize video pipeline on startup to avoid blocking