Update skyreelsinfer/pipelines/pipeline_skyreels_video.py

skyreelsinfer/pipelines/pipeline_skyreels_video.py

@@ -14,7 +14,7 @@ from diffusers.pipelines.hunyuan_video.pipeline_hunyuan_video import MultiPipeli
 from diffusers.pipelines.hunyuan_video.pipeline_hunyuan_video import PipelineCallback
 from diffusers.pipelines.hunyuan_video.pipeline_hunyuan_video import retrieve_timesteps
 from PIL import Image
-#import gc
+import gc
 
 def resizecrop(image, th, tw):
     w, h = image.size
@@ -339,7 +339,9 @@ class SkyreelsVideoPipeline(HunyuanVideoPipeline):
 
         if hasattr(self, "text_encoder_to_cpu"):
             self.text_encoder_to_cpu()
-        self.text_encoder.to("cpu")
+        #self.text_encoder.to("cpu")
+        del self.text_encoder
+        gc.collect()
         self.vae.to("cpu")
         torch.cuda.empty_cache()
         torch.cuda.reset_peak_memory_stats()