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