From cc6f662a54c93275499982ffa49cf475eb2328ec Mon Sep 17 00:00:00 2001 From: Matt Date: Tue, 14 Jan 2025 18:24:57 +0000 Subject: [PATCH] Testing success, remove debug block --- src/transformers/modeling_utils.py | 6 ------ 1 file changed, 6 deletions(-) diff --git a/src/transformers/modeling_utils.py b/src/transformers/modeling_utils.py index c769de0fb..f98b3c6e8 100755 --- a/src/transformers/modeling_utils.py +++ b/src/transformers/modeling_utils.py @@ -580,12 +580,6 @@ def set_initialized_submodules(model, state_dict_keys): # Iterate until we reach the end of the keys with this prefix loaded_keys.add(state_dict_keys[i].removeprefix(prefix)) i += 1 - # This next block is just for debug testing and should be removed before merging - old_loaded_keys = { - k.removeprefix(f"{module_name}.") for k in state_dict_keys if k.startswith(f"{module_name}.") - } - if not old_loaded_keys == loaded_keys: - breakpoint() # When checking if the root module is loaded all state_dict_keys must be used. if module_name == "": loaded_keys = set(state_dict_keys)