mirror of
https://github.com/huggingface/lerobot.git
synced 2026-07-25 10:46:01 +00:00
precommit
This commit is contained in:
@@ -40,7 +40,7 @@ lerobot-rollout \
|
||||
```
|
||||
|
||||
| Flag | Description |
|
||||
| ---------------- | ------------------------------------------------------ |
|
||||
| ---------------- | ------------------------------------------------------------------------------------------------------------------------------------------- |
|
||||
| `--duration` | Run time in seconds (0 = infinite) |
|
||||
| `--task` | Task description passed to the policy |
|
||||
| `--display_data` | Stream observations/actions to the visualization backend (`--display_mode`; add `--display_extra_data` for a policy's imagined predictions) |
|
||||
@@ -242,7 +242,7 @@ See the [Real-Time Chunking](./rtc) guide for details on tuning RTC parameters.
|
||||
## Common Flags
|
||||
|
||||
| Flag | Description | Default |
|
||||
| --------------------------------- | ----------------------------------------------------------------- | ------- |
|
||||
| --------------------------------- | ----------------------------------------------------------------------------------------------------------------------------------------------------------- | ------- |
|
||||
| `--policy.path` | **Required.** HF Hub model ID or local checkpoint path | -- |
|
||||
| `--robot.type` | **Required.** Robot type (e.g. `so100_follower`, `koch_follower`) | -- |
|
||||
| `--robot.port` | Serial port for the robot | -- |
|
||||
|
||||
@@ -419,7 +419,9 @@ class LingBotVAPolicy(PreTrainedPolicy):
|
||||
# First call: this observation conditions the first chunk (it is *not* a keyframe).
|
||||
self._started = True
|
||||
actions, predictions = unpack_action_output(
|
||||
self.predict_action_chunk(batch, return_intermediate_predictions=return_intermediate_predictions)
|
||||
self.predict_action_chunk(
|
||||
batch, return_intermediate_predictions=return_intermediate_predictions
|
||||
)
|
||||
) # [B, chunk_size, n_used]
|
||||
self._action_queue.extend(actions.transpose(0, 1)) # [chunk_size, B, n_used]
|
||||
self._obs_buffer = []
|
||||
|
||||
Reference in New Issue
Block a user