fix config issue blocking CI

Signed-off-by: Xin He <[email protected]>
intel · Dec 12, 2024 · d823429 · d823429
1 parent b674850
commit d823429
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/neural_compressor/torch/quantization/config.py b/neural_compressor/torch/quantization/config.py
@@ -1955,7 +1955,7 @@ def get_default_fp8_config_set() -> FP8Config:
 
 ######################## MixedPrecision Config ###############################
 @register_config(framework_name=FRAMEWORK_NAME, algo_name=MIXED_PRECISION)
-class MixedPrecisionConfig(BaseConfig):
+class MixedPrecisionConfig(TorchBaseConfig):
     """Config class for mixed-precision."""
 
     name = MIXED_PRECISION
@@ -2080,7 +2080,7 @@ def get_woq_tuning_config() -> list:
 )
 
 
-class StaticQuantConfig:
+class StaticQuantConfig(TorchBaseConfig):
     _model_mapping = CONFIGS_FOR_STATIC_QUANT_MAPPING
 
     def __new__(