[Auto Merge] test: use CPU offload to save GPU memory in multi-batch test (#344)

tests/flux/test_multiple_batch.py

@@ -11,7 +11,7 @@ from .utils import run_test
     "height,width,attention_impl,cpu_offload,expected_lpips,batch_size",
     [
         (1024, 1024, "nunchaku-fp16", False, 0.140 if get_precision() == "int4" else 0.135, 2),
-        (1920, 1080, "flashattn2", False, 0.160 if get_precision() == "int4" else 0.123, 4),
+        (1920, 1080, "flashattn2", True, 0.160 if get_precision() == "int4" else 0.123, 4),
     ],
 )
 def test_flux_schnell(