Update handler.py

handler.py

@@ -90,6 +90,11 @@ class EndpointHandler:
         
         # Initialize VAE in float16
         self.pipeline.vae = self.pipeline.vae.half()
+
+        # Initialize Enhance-A-Video
+        inject_enhance_for_hunyuanvideo(self.pipeline.transformer)
+        set_enhance_weight(4.0)  # default weight
+        enable_enhance()
         
         # Initialize Varnish for post-processing
         self.varnish = Varnish(
@@ -166,6 +171,14 @@ class EndpointHandler:
             else:
                 disable_teacache(self.pipeline.transformer)
 
+            # Configure Enhance-A-Video weight if enabled
+            if config.enable_enhance_a_video:
+                set_enhance_weight(config.enhance_a_video_weight)
+                enable_enhance()
+            else:
+                # Reset enhance weight to 0 to effectively disable it
+                set_enhance_weight(0)
+
             # Generate video frames
             with torch.inference_mode():
                 output = self.pipeline(