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

[pre-commit.ci] auto fixes from pre-commit.com hooks

Browse Source 
for more information, see https://pre-commit.ci
This commit is contained in:
pre-commit-ci[bot]
2025-06-03 13:32:36 +00:00
parent f01c6d06cf
commit 97f061fca6
1 changed files with 7 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

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

@@ -656,8 +656,13 @@ class VLAFlowMatching(nn.Module):
dtype = action_emb.dtype dtype = action_emb.dtype
# Embed timestep using sine-cosine positional encoding with sensitivity in the range [0, 1] # Embed timestep using sine-cosine positional encoding with sensitivity in the range [0, 1]
time_emb = create_sinusoidal_pos_embedding( time_emb = create_sinusoidal_pos_embedding(
timestep, self.vlm_with_expert.expert_hidden_size, self.config.min_period, self.config.max_period, device=device) timestep,
self.vlm_with_expert.expert_hidden_size,
self.config.min_period,
self.config.max_period,
device=device,
)
time_emb = time_emb.type(dtype=dtype) time_emb = time_emb.type(dtype=dtype)
time_emb = time_emb[:, None, :].expand_as(action_emb) time_emb = time_emb[:, None, :].expand_as(action_emb)