Update skyreelsinfer/pipelines/pipeline_skyreels_video.py

skyreelsinfer/pipelines/pipeline_skyreels_video.py

@@ -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,7 +344,6 @@ class SkyreelsVideoPipeline(HunyuanVideoPipeline):
 
         if hasattr(self, "text_encoder_to_cpu"):
             self.text_encoder_to_cpu()
-        self.text_encoder.to("cpu")
         self.vae.to("cpu")
         torch.cuda.empty_cache()
         torch.cuda.reset_peak_memory_stats()