hangX/lerobot
1
0
Fork 0
forked from tangger/lerobot
Code Pull Requests 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:
```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,