From 44b457852eb5a3d17296ab8ca5deed3fc284e461 Mon Sep 17 00:00:00 2001 From: mshukor Date: Wed, 28 May 2025 21:31:21 +0200 Subject: [PATCH] precommit fix --- lerobot/common/policies/smolvla/modeling_smolvla.py | 4 +--- 1 file changed, 1 insertion(+), 3 deletions(-) diff --git a/lerobot/common/policies/smolvla/modeling_smolvla.py b/lerobot/common/policies/smolvla/modeling_smolvla.py index 4c3a17c32..b399ae1d3 100644 --- a/lerobot/common/policies/smolvla/modeling_smolvla.py +++ b/lerobot/common/policies/smolvla/modeling_smolvla.py @@ -43,7 +43,7 @@ python lerobot/scripts/train.py \ Example of using the smolvla pretrained model outside LeRobot training framework: ```python -policy = SmolVLAPolicy.from_pretrained("lerobot/smolvla") +policy = SmolVLAPolicy.from_pretrained("lerobot/smolvla_base") ``` """ @@ -551,11 +551,9 @@ class VLAFlowMatching(nn.Module): """Embed images with SigLIP and language tokens with embedding layer to prepare for SmolVLM transformer processing. """ - # TODO: avoid list in python and torch.cat ; prefer pre-allocation with torch.empty embs = [] pad_masks = [] att_masks = [] - # TODO: remove for loop for _img_idx, ( img, img_mask,