fix(policies): add device back to smolvlm expert (#2662)

This commit is contained in:
Steven Palma
2025-12-17 12:12:03 +01:00
committed by GitHub
parent 2b304eeb84
commit b1d162f333
@@ -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