huangyanpei/VLA
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Fixed the bug in flow matching (#68)

Browse Source 
* fix a bug in flow matching

* fixed the bugs in flow matching

* fixed format

---------

Co-authored-by: Geoffrey19 <sympathischmann35@gmail.com>
This commit is contained in:
Tong Wu
2025-11-07 15:11:49 +08:00
committed by GitHub
co-authored by Geoffrey19
parent fbe354c266
commit 7b57b7800b
2 changed files with 2 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files
wall_x/model/action_head.py
+1 -1
View File
@@ -307,7 +307,7 @@ class ActionProcessor(nn.Module):
torch.Tensor: Sampled timesteps of shape [batch_size] torch.Tensor: Sampled timesteps of shape [batch_size]
""" """
sample = self.beta_dist.sample([batch_size]).to(device=device, dtype=dtype) sample = self.beta_dist.sample([batch_size]).to(device=device, dtype=dtype)
time = (1 - sample) / self.s time = (1 - sample) * self.s
return time return time
def proprioception_proj( def proprioception_proj(
wall_x/model/qwen2_5_based/modeling_qwen2_5_vl_act.py
+1 -1
View File
@@ -2046,7 +2046,7 @@ class Qwen2_5_VLMoEForAction(Qwen2_5_VLForConditionalGeneration):
times = torch.linspace( times = torch.linspace(
0, 0,
1, 1,
num_inference_timesteps, num_inference_timesteps + 1,
device=inputs_embeds.device, device=inputs_embeds.device,
dtype=inputs_embeds.dtype, dtype=inputs_embeds.dtype,
) )