From 492b1812ddafc425e838f036db61ee68657d1fdb Mon Sep 17 00:00:00 2001
From: fracapuano <francesco.capuano@huggingface.co>
Date: Mon, 2 Jun 2025 11:23:40 +0200
Subject: [PATCH] fix: device to send VLM to is found automatically

---
 lerobot/common/policies/smolvla/smolvlm_with_expert.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/lerobot/common/policies/smolvla/smolvlm_with_expert.py b/lerobot/common/policies/smolvla/smolvlm_with_expert.py
index ef369969..90b70506 100644
--- a/lerobot/common/policies/smolvla/smolvlm_with_expert.py
+++ b/lerobot/common/policies/smolvla/smolvlm_with_expert.py
@@ -77,7 +77,7 @@ class SmolVLMWithExpertModel(nn.Module):
             print(f"Loading  {model_id} weights ...")
             self.vlm = AutoModelForImageTextToText.from_pretrained(
                 model_id,
-                device_map="cuda",
+                device_map="auto",
                 torch_dtype="bfloat16",
                 low_cpu_mem_usage=True,
             )