tangger/lerobot
1
0
Fork 1
Code Issues Pull Requests Actions 1 Packages Projects Releases Wiki Activity

precommit fix

Browse Source
This commit is contained in:
mshukor
2025-05-28 21:31:21 +02:00
parent 97d609feb4
commit 44b457852e
1 changed files with 1 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
lerobot/common/policies/smolvla/modeling_smolvla.py
View File

@@ -43,7 +43,7 @@ python lerobot/scripts/train.py \
Example of using the smolvla pretrained model outside LeRobot training framework: Example of using the smolvla pretrained model outside LeRobot training framework:
```python ```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 """Embed images with SigLIP and language tokens with embedding layer to prepare
for SmolVLM transformer processing. for SmolVLM transformer processing.
""" """
# TODO: avoid list in python and torch.cat ; prefer pre-allocation with torch.empty
embs = [] embs = []
pad_masks = [] pad_masks = []
att_masks = [] att_masks = []
# TODO: remove for loop
for _img_idx, ( for _img_idx, (
img, img,
img_mask, img_mask,