Update handler.py

handler.py

@@ -410,20 +410,24 @@ class EndpointHandler:
                     if self._current_lora_model != (config.lora_model_name, config.lora_model_weight_file):
                         # Unload previous LoRA if it exists and is different
                         if hasattr(self.text_to_video, 'unload_lora_weights'):
+                            print("Unloading LoRA weights for the text_to_video pipeline..")
                             self.text_to_video.unload_lora_weights()
                         
                         if support_image_prompt and hasattr(self.image_to_video, 'unload_lora_weights'):
+                            print("Unloading LoRA weights for the image_to_video pipeline..")
                             self.image_to_video.unload_lora_weights()
     
                 if config.lora_model_name:
                     # Load new LoRA
                     if hasattr(self.text_to_video, 'load_lora_weights'):
+                        print("Loading LoRA weights for the text_to_video pipeline..")
                         self.text_to_video.load_lora_weights(
                             config.lora_model_name,
                             weight_name=config.lora_model_weight_file if config.lora_model_weight_file else None,
                             token=hf_token,
                         )
                     if support_image_prompt and hasattr(self.image_to_video, 'load_lora_weights'):
+                        print("Loading LoRA weights for the image_to_video pipeline..")
                         self.image_to_video.load_lora_weights(
                             config.lora_model_name,
                             weight_name=config.lora_model_weight_file if config.lora_model_weight_file else None,
@@ -435,15 +439,15 @@ class EndpointHandler:
                 if config.lora_model_trigger:
                     generation_kwargs["prompt"] = f"{config.lora_model_trigger} {generation_kwargs['prompt']}"
 
-                enhance_a_video_config = EnhanceAVideoConfig(
-                    weight=config.enhance_a_video_weight if config.enable_enhance_a_video else 0.0,
-                    # doing some testing
-                    num_frames_callback=lambda: (8 + 1),
-                    # num_frames_callback=lambda: config.num_frames,
-                    # num_frames_callback=lambda: (config.num_frames - 1),
-                    
-                    _attention_type=1
-                )
+                #enhance_a_video_config = EnhanceAVideoConfig(
+                #    weight=config.enhance_a_video_weight if config.enable_enhance_a_video else 0.0,
+                #    # doing some testing
+                #    num_frames_callback=lambda: (8 + 1),
+                #    # num_frames_callback=lambda: config.num_frames,
+                #    # num_frames_callback=lambda: (config.num_frames - 1),
+                #    
+                #    _attention_type=1
+                #)
                 
                 # Check if image-to-video generation is requested
                 if support_image_prompt and input_image: