mirror of
https://github.com/huggingface/lerobot.git
synced 2026-05-22 03:59:42 +00:00
Debug
This commit is contained in:
@@ -270,7 +270,7 @@ class RTCProcessor:
|
|||||||
execution_horizon=execution_horizon,
|
execution_horizon=execution_horizon,
|
||||||
)
|
)
|
||||||
|
|
||||||
return result
|
return result, x_t
|
||||||
|
|
||||||
def get_prefix_weights(self, start, end, total):
|
def get_prefix_weights(self, start, end, total):
|
||||||
start = min(start, end)
|
start = min(start, end)
|
||||||
|
|||||||
@@ -806,7 +806,7 @@ class VLAFlowMatching(nn.Module):
|
|||||||
prev_chunk_left_over = kwargs.get("prev_chunk_left_over")
|
prev_chunk_left_over = kwargs.get("prev_chunk_left_over")
|
||||||
execution_horizon = kwargs.get("execution_horizon")
|
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,
|
x_t=x_t,
|
||||||
prev_chunk_left_over=prev_chunk_left_over,
|
prev_chunk_left_over=prev_chunk_left_over,
|
||||||
inference_delay=inference_delay,
|
inference_delay=inference_delay,
|
||||||
|
|||||||
Reference in New Issue
Block a user