Update save_load.py

intel · Jan 9, 2025 · c6d75da · c6d75da
1 parent 1c0fabb
commit c6d75da
Showing 1 changed file with 0 additions and 2 deletions.
diff --git a/neural_compressor/torch/algorithms/weight_only/save_load.py b/neural_compressor/torch/algorithms/weight_only/save_load.py
@@ -77,8 +77,6 @@ def save(model, output_dir="./saved_results", format=SaveLoadFormat.DEFAULT, **k
     # saving process
     save_config_mapping(model.qconfig, qconfig_file_path)
 
-    # MethodType 'save' not in state_dict
-    del model.save
     torch.save(model.state_dict(), qmodel_weight_file_path)
 
     logger.info("Save quantized model weight to {}.".format(qmodel_weight_file_path))