Fix traacking

This commit is contained in:
Eugene Mironov
2025-11-06 02:30:53 +07:00
parent 55c4cc1b27
commit b27570039c
4 changed files with 11 additions and 4 deletions
@@ -40,7 +40,7 @@ class RTCConfig:
# Core RTC settings
# Todo change to exp
prefix_attention_schedule: RTCAttentionSchedule = RTCAttentionSchedule.LINEAR
max_guidance_weight: float = 5.0
max_guidance_weight: float = 10.0
execution_horizon: int = 10
# Debug settings
-1
View File
@@ -260,7 +260,6 @@ class RTCProcessor:
x1_t = x1_t.squeeze(0)
err = err.squeeze(0)
# Record debug information (all params except x_t which is recorded externally)
self.track(
time=time,
x1_t=x1_t,
@@ -798,12 +798,13 @@ class VLAFlowMatching(nn.Module):
# Euler step
x_t += dt * v_t
time += dt
# Record x_t and v_t after Euler step (other params are recorded in rtc_processor.denoise_step)
if self.rtc_processor is not None and self.rtc_processor.is_debug_enabled():
self.rtc_processor.track(time=time, x_t=x_t, v_t=v_t)
time += dt
return x_t
def denoise_step(