Skip to content

Commit

Permalink
remove SDXL test
Browse files Browse the repository at this point in the history
  • Loading branch information
l-bat committed Mar 10, 2024
1 parent 3544c4b commit 067c6d5
Showing 1 changed file with 1 addition and 2 deletions.
3 changes: 1 addition & 2 deletions tests/openvino/test_quantization.py
Original file line number Diff line number Diff line change
Expand Up @@ -236,7 +236,6 @@ class OVWeightCompressionTest(unittest.TestCase):

SUPPORTED_ARCHITECTURES_WITH_HYBRID_QUANTIZATION = (
(OVStableDiffusionPipeline, "stable-diffusion", 72, 195),
(OVStableDiffusionXLPipeline, "stable-diffusion-xl", 84, 331),
(OVLatentConsistencyModelPipeline, "latent-consistency", 50, 135),
)

Expand Down Expand Up @@ -373,7 +372,7 @@ def test_ovmodel_hybrid_quantization(self, model_cls, model_type, expected_num_f

model.save_pretrained(tmp_dir)

@parameterized.expand(SUPPORTED_ARCHITECTURES_WITH_HYBRID_QUANTIZATION[2:])
@parameterized.expand(SUPPORTED_ARCHITECTURES_WITH_HYBRID_QUANTIZATION)
def test_ovmodel_hybrid_quantization_with_custom_dataset(
self, model_cls, model_type, expected_num_fake_quantize, expected_ov_int8
):
Expand Down

0 comments on commit 067c6d5

Please sign in to comment.