Update packages/ltx-pipelines/src/ltx_pipelines/keyframe_interpolation.py

packages/ltx-pipelines/src/ltx_pipelines/keyframe_interpolation.py

@@ -159,7 +159,7 @@ class KeyframeInterpolationPipeline:
         )
 
         torch.cuda.synchronize()
-        # del transformer
+        del transformer
         utils.cleanup_memory()
 
         # Stage 2: Upsample and refine the video at higher resolution with distilled LORA.
@@ -218,8 +218,8 @@ class KeyframeInterpolationPipeline:
         )
 
         torch.cuda.synchronize()
-        # del transformer
-        # del video_encoder
+        del transformer
+        del video_encoder
         utils.cleanup_memory()
 
         decoded_video = vae_decode_video(video_state, self.stage_2_model_ledger.video_decoder(), tiling_config)