fix typo

Signed-off-by: changwangss <[email protected]>
intel · Jun 19, 2024 · dcf3369 · dcf3369
1 parent 01649dd
commit dcf3369
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/docs/quantization.md b/docs/quantization.md
@@ -134,12 +134,12 @@ Quantization methods include the following three types:
 ## Get Started
 ### Script:
 ```python
-from neural_compressor.config import PostTrainingConfig
+from neural_compressor.config import PostTrainingQuantConfig
 from intel_extension_for_transformers.transformers.trainer import NLPTrainer
 # Replace transformers.Trainer with NLPTrainer
 # trainer = transformers.Trainer(......)
 trainer = NLPTrainer(......)
-q_config = PostTrainingConfig(
+q_config = PostTrainingQuantConfig(
     approach="static"
 )
 model = trainer.quantize(quant_config=q_config)