Update skyreelsinfer/skyreels_video_infer.py

skyreelsinfer/skyreels_video_infer.py

@@ -45,7 +45,7 @@ class SkyReelsVideoSingleGpuInfer:
             # subfolder="transformer",
             torch_dtype=torch.bfloat16,
             device="cpu",
-        ).to("cpu")
+        ).to("cpu").eval()
         if quant_model:
             quantize_(text_encoder, float8_weight_only(), device="cpu")
             text_encoder.to("cpu")