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

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

View File

@@ -656,8 +656,13 @@ class VLAFlowMatching(nn.Module):
dtype = action_emb.dtype
# Embed timestep using sine-cosine positional encoding with sensitivity in the range [0, 1]
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[:, None, :].expand_as(action_emb)