# skip this test # 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, # )