assume always there is state

2025-06-02 20:50:38 +02:00
parent 1fd53b441a
commit c2d645717a
1 changed files with 10 additions and 11 deletions
--- a/lerobot/common/policies/smolvla/modeling_smolvla.py
+++ b/lerobot/common/policies/smolvla/modeling_smolvla.py
@@ -616,7 +616,6 @@ class VLAFlowMatching(nn.Module):
        num_lang_embs = lang_emb.shape[1]
        att_masks += [0] * num_lang_embs

-        if state is not None:
        state_emb = self.state_proj(state)
        state_emb = state_emb[:, None, :] if state_emb.ndim == 2 else state_emb
        embs.append(state_emb)