Update skyreelsinfer/skyreels_video_infer.py

skyreelsinfer/skyreels_video_infer.py

@@ -49,10 +49,10 @@ class SkyReelsVideoSingleGpuInfer:
         if quant_model:
             quantize_(text_encoder, float8_weight_only(), device="cpu")
             text_encoder.to("cpu")
-            torch.cuda.empty_cache()
+            #torch.cuda.empty_cache()
             quantize_(transformer, float8_weight_only(), device="cpu")
             transformer.to("cpu")
-            torch.cuda.empty_cache()
+            #torch.cuda.empty_cache()
         pipe = SkyreelsVideoPipeline.from_pretrained(
             base_model_id,
             transformer=transformer,