From 12eef5bb83b43d09c59b08592e37a489ff1cb86d Mon Sep 17 00:00:00 2001
From: kijai <40791699+kijai@users.noreply.github.com>
Date: Tue, 25 Mar 2025 11:08:47 +0200
Subject: [PATCH] Disable enhance-a-video when sampling single image

---
 nodes.py | 3 ++-
 1 file changed, 2 insertions(+), 1 deletion(-)

diff --git a/nodes.py b/nodes.py
index a23ee1d..0630cfc 100644
--- a/nodes.py
+++ b/nodes.py
@@ -1812,7 +1812,7 @@ class WanVideoSampler:
         elif model["manual_offloading"]:
             transformer.to(device)
         #feta
-        if feta_args is not None:
+        if feta_args is not None and latent_video_length > 1:
             set_enhance_weight(feta_args["weight"])
             feta_start_percent = feta_args["start_percent"]
             feta_end_percent = feta_args["end_percent"]
@@ -1822,6 +1822,7 @@ class WanVideoSampler:
                 set_num_frames(latent_video_length)
             enable_enhance()
         else:
+            feta_args = None
             disable_enhance()
 
         # Initialize TeaCache if enabled