admin/lerobot
1
0
Fork 0
mirror of https://github.com/huggingface/lerobot.git synced 2026-07-24 18:26:11 +00:00
Code Issues Packages Projects Releases Wiki Activity

migrate policy revert

Browse Source
This commit is contained in:
Jade Choghari
2025-11-17 14:05:09 +01:00
parent 5277a9909d
commit 858626dea5
1 changed files with 0 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files
src/lerobot/processor/migrate_policy_normalization.py
-3
View File
@@ -325,9 +325,6 @@ def load_state_dict_with_missing_key_handling(
Returns: Returns:
List of problematic missing keys that weren't in the whitelist. 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 the cleaned state dict with strict=False to capture missing/unexpected keys
load_result = policy.load_state_dict(state_dict, strict=False) load_result = policy.load_state_dict(state_dict, strict=False)