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
@@ -311,6 +311,11 @@ class SkyreelsVideoPipeline(HunyuanVideoPipeline):
             generator,
             latents,
         )
+        
+        self.text_encoder.to("cpu")
+        torch.cuda.empty_cache()
+        torch.cuda.reset_peak_memory_stats()  
+        
         # add image latents
         if image is not None:
             image_latents = self.image_latents(
@@ -339,9 +344,6 @@ class SkyreelsVideoPipeline(HunyuanVideoPipeline):
 
         if hasattr(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()