From 858626dea519fa247617f839ceab204e79ad280a Mon Sep 17 00:00:00 2001 From: Jade Choghari Date: Mon, 17 Nov 2025 14:05:09 +0100 Subject: [PATCH] migrate policy revert --- src/lerobot/processor/migrate_policy_normalization.py | 3 --- 1 file changed, 3 deletions(-) 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)