This commit is contained in:
Pepijn
2026-01-20 20:08:28 +01:00
parent 0cb8c92fe4
commit f4ccf911fa
7 changed files with 5 additions and 18 deletions
@@ -48,4 +48,3 @@ def test_apply_rtc_training_time_prefix_mask():
# Delay=2 means the first two steps are prefix (time forced to 0.0) and only the last two are postfix.
assert torch.allclose(time_tokens[0], torch.tensor([0.0, 0.0, 0.5, 0.5]))
assert torch.equal(postfix_mask[0], torch.tensor([False, False, True, True]))