Update handler.py

handler.py

@@ -47,6 +47,11 @@ class GenerationConfig:
     enable_teacache: bool = True
     teacache_threshold: float = 0.15
 
+    
+    # Enhance-A-Video settings
+    enable_enhance_a_video: bool = True
+    enhance_a_video_weight: float = 4.0
+
     def validate_and_adjust(self) -> 'GenerationConfig':
         """Validate and adjust parameters"""
         # Ensure num_frames follows 4k + 1 format
@@ -134,7 +139,9 @@ class EndpointHandler:
             audio_prompt=params.get("audio_prompt", ""),
             audio_negative_prompt=params.get("audio_negative_prompt", "voices, voice, talking, speaking, speech"),
             enable_teacache=params.get("enable_teacache", True),
-            teacache_threshold=params.get("teacache_threshold", 0.15)
+            teacache_threshold=params.get("teacache_threshold", 0.15),
+            enable_enhance_a_video=params.get("enable_enhance_a_video", True),
+            enhance_a_video_weight=params.get("enhance_a_video_weight", 4.0)
         ).validate_and_adjust()
 
         try:
@@ -209,8 +216,10 @@ class EndpointHandler:
                         "fps": result.metadata.fps,
                         "duration": result.metadata.duration,
                         "seed": config.seed,
-                        "teacache_enabled": config.enable_teacache,
+                        "enable_teacache": config.enable_teacache,
                         "teacache_threshold": config.teacache_threshold if config.enable_teacache else 0,
+                        "enable_enhance_a_video": config.enable_enhance_a_video,
+                        "enhance_a_video_weight": config.enhance_a_video_weight if config.enable_enhance_a_video else 0,
                     }
                 }