From b1d162f333d4a7e15151e499234e9c3a8e5136ab Mon Sep 17 00:00:00 2001 From: Steven Palma Date: Wed, 17 Dec 2025 12:12:03 +0100 Subject: [PATCH] fix(policies): add device back to smolvlm expert (#2662) --- src/lerobot/policies/smolvla/modeling_smolvla.py | 1 + 1 file changed, 1 insertion(+) diff --git a/src/lerobot/policies/smolvla/modeling_smolvla.py b/src/lerobot/policies/smolvla/modeling_smolvla.py index cce41def8..485d3e4e5 100644 --- a/src/lerobot/policies/smolvla/modeling_smolvla.py +++ b/src/lerobot/policies/smolvla/modeling_smolvla.py @@ -527,6 +527,7 @@ class VLAFlowMatching(nn.Module): num_vlm_layers=self.config.num_vlm_layers, self_attn_every_n_layers=self.config.self_attn_every_n_layers, expert_width_multiplier=self.config.expert_width_multiplier, + device=self.config.device if self.config.device is not None else "auto", ) self.state_proj = nn.Linear( self.config.max_state_dim, self.vlm_with_expert.config.text_config.hidden_size