Khalil Meftah 46044fed75 Fix: remove device_map from SmolVLA model loading (#3029) ...

* Fix SmolVLA meta tensor error by removing device_map

- Remove device_map parameter from VLM model loading
- Change torch_dtype from string to torch.bfloat16
- Add explicit .to(device) calls after initialization

This resolves NotImplementedError when training SmolVLA policy.
Fixes meta tensor copy issue in factory.py:418.

* fix: remove manual device movement logic and fix dtype handling

---------

Co-authored-by: Highsky7 <albert31115@gmail.com>

2026-02-26 13:28:46 +01:00

..

lerobot

Fix: remove device_map from SmolVLA model loading (#3029)

2026-02-26 13:28:46 +01:00