Update app.py

app.py

@@ -35,6 +35,8 @@ hf_hub_download_local(repo_id="Kijai/WanVideo_comfy", filename="Wan22-Lightning/
 hf_hub_download_local(repo_id="Kijai/WanVideo_comfy", filename="Wan22-Lightning/Wan2.2-Lightning_I2V-A14B-4steps-lora_LOW_fp16.safetensors", local_dir="models/loras")
 print("Downloads complete.")
 
+model_management.vram_state = model_management.VRAMState.HIGH_VRAM
+
 # --- Image Processing Functions ---
 def calculate_video_dimensions(width, height, max_size=832, min_size=480):
     """