diff --git a/src/lerobot/processor/migrate_policy_normalization.py b/src/lerobot/processor/migrate_policy_normalization.py index e602bb5d0..525b7431c 100644 --- a/src/lerobot/processor/migrate_policy_normalization.py +++ b/src/lerobot/processor/migrate_policy_normalization.py @@ -325,9 +325,6 @@ def load_state_dict_with_missing_key_handling( Returns: List of problematic missing keys that weren't in the whitelist. """ - state_dict["model.vlm.language_model.model.encoder.embed_tokens.weight"] = state_dict[ - "model.vlm.language_model.model.shared.weight" - ].clone() # Load the cleaned state dict with strict=False to capture missing/unexpected keys load_result = policy.load_state_dict(state_dict, strict=False)