diff --git a/lerobot/common/policies/smolvla/smolvlm_with_expert.py b/lerobot/common/policies/smolvla/smolvlm_with_expert.py index 90b705062..ce3b5cd04 100644 --- a/lerobot/common/policies/smolvla/smolvlm_with_expert.py +++ b/lerobot/common/policies/smolvla/smolvlm_with_expert.py @@ -373,7 +373,6 @@ class SmolVLMWithExpertModel(nn.Module): ] # take into account kv expert_query_states = apply_rope(expert_query_state, expert_position_id) - # expert_key_states = apply_rope(expert_key_state, expert_position_id) att_output = attention_interface( expert_attention_mask,