Update handler.py

handler.py

@@ -12,6 +12,7 @@ import traceback
 import os
 import numpy as np
 import torch
+import gc
 
 from diffusers import LTXPipeline, LTXImageToVideoPipeline
 from diffusers.hooks import apply_enhance_a_video, EnhanceAVideoConfig
@@ -458,6 +459,10 @@ class EndpointHandler:
                     asyncio.set_event_loop(loop)
                 
                 video_uri, metadata = loop.run_until_complete(self.process_frames(frames, config))
+
+                torch.cuda.empty_cache()
+                torch.cuda.reset_peak_memory_stats()
+                gc.collect()
                 
                 return {
                     "video": video_uri,