From 82196712ea95806375712abbf3ef13cb53fb0806 Mon Sep 17 00:00:00 2001 From: Sara Adkins Date: Wed, 29 May 2024 16:58:55 +0000 Subject: [PATCH] fix quant modifier name --- .../export/transformers/test_generative_transformers.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/tests/sparseml/export/transformers/test_generative_transformers.py b/tests/sparseml/export/transformers/test_generative_transformers.py index 1a9af341cb9..d09b5e520d8 100644 --- a/tests/sparseml/export/transformers/test_generative_transformers.py +++ b/tests/sparseml/export/transformers/test_generative_transformers.py @@ -131,7 +131,7 @@ def test_export_with_recipe(self, setup): recipe = """test_stage: quant_modifiers: - QuantizationModifier: + LegacyQuantizationModifier: post_oneshot_calibration: False scheme_overrides: Embedding: