rm loaded_state_dict_keys params

Signed-off-by: Kaihui-intel <[email protected]>
intel · Sep 30, 2024 · e9404b5 · e9404b5
1 parent 72398b6
commit e9404b5
Showing 1 changed file with 0 additions and 1 deletion.
diff --git a/neural_compressor/torch/algorithms/weight_only/save_load.py b/neural_compressor/torch/algorithms/weight_only/save_load.py
@@ -837,7 +837,6 @@ def _load_remaining_pretrained_weight(self, model):
             _load_state_dict_into_meta_model(
                 model=model,
                 state_dict=state_dict,
-                loaded_state_dict_keys=self.loaded_state_dict_keys,
                 start_prefix="",
                 expected_keys=list(state_dict.keys()),
                 device_map={"": self.device},