Fixed the bug in flow matching (#68)

* 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
@@ -2046,7 +2046,7 @@ class Qwen2_5_VLMoEForAction(Qwen2_5_VLForConditionalGeneration):
times = torch.linspace(
0,
1,
num_inference_timesteps,
num_inference_timesteps + 1,
device=inputs_embeds.device,
dtype=inputs_embeds.dtype,
)