fix CI

Signed-off-by: Mengni Wang <[email protected]>
onnx · Sep 19, 2024 · d455061 · d455061
1 parent b51d8ec
commit d455061
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 2 deletions.
diff --git a/onnx_neural_compressor/algorithms/utility.py b/onnx_neural_compressor/algorithms/utility.py
@@ -621,6 +621,7 @@ def quant_matmul_weight_only(
         new_inits.append(q_weight_tensor)
     return new_nodes, new_inits, remove_nodes
 
+
 def prepare_inputs(model, data_reader, providers):
     """Prepare inputs for weight only quantization.
 

diff --git a/onnx_neural_compressor/quantization/config.py b/onnx_neural_compressor/quantization/config.py
@@ -1640,7 +1640,7 @@ def to_config_mapping(self, model: Union[onnx.ModelProto, str], config_list: Lis
             model = onnx.load(model, load_external_data=False)
 
         model_info = BaseConfig.get_model_info(model)
- 
+
         if config_list is None:
             config_list = [self]
         for config in config_list:
@@ -2015,7 +2015,7 @@ def to_config_mapping(self, model: Union[onnx.ModelProto, str], config_list: Lis
             model = onnx.load(model, load_external_data=False)
 
         model_info = BaseConfig.get_model_info(model)
- 
+
         if config_list is None:
             config_list = [self]
         for config in config_list: