This commit is contained in:
Eugene Mironov
2025-11-07 02:22:47 +07:00
parent 79ffe316e4
commit 7baf909e32
2 changed files with 2 additions and 2 deletions
+1 -1
View File
@@ -270,7 +270,7 @@ class RTCProcessor:
execution_horizon=execution_horizon,
)
return result
return result, x_t
def get_prefix_weights(self, start, end, total):
start = min(start, end)
@@ -806,7 +806,7 @@ class VLAFlowMatching(nn.Module):
prev_chunk_left_over = kwargs.get("prev_chunk_left_over")
execution_horizon = kwargs.get("execution_horizon")
v_t = self.rtc_processor.denoise_step(
v_t, x_t = self.rtc_processor.denoise_step(
x_t=x_t,
prev_chunk_left_over=prev_chunk_left_over,
inference_delay=inference_delay,