pytest test multiple batches

tests/flux/test_multiple_batch.py

+import pytest
+
+from nunchaku.utils import get_precision, is_turing
+from .utils import run_test
+
+
+@pytest.mark.skipif(is_turing(), reason="Skip tests due to using Turing GPUs")
+@pytest.mark.parametrize(
+    "height,width,attention_impl,cpu_offload,expected_lpips,batch_size",
+    [
+        (1024, 1024, "nunchaku-fp16", False, 0.126, 2),
+        (1920, 1080, "flashattn2", False, 0.141, 4),
+    ],
+)
+def test_int4_schnell(
+    height: int, width: int, attention_impl: str, cpu_offload: bool, expected_lpips: float, batch_size: int
+):
+    run_test(
+        precision=get_precision(),
+        height=height,
+        width=width,
+        attention_impl=attention_impl,
+        cpu_offload=cpu_offload,
+        expected_lpips=expected_lpips,
+        batch_size=batch_size,
+    )