mirror of
https://github.com/huggingface/lerobot.git
synced 2026-07-24 02:06:15 +00:00
fix circular input
This commit is contained in:
@@ -16,9 +16,6 @@
|
|||||||
|
|
||||||
from dataclasses import dataclass, field
|
from dataclasses import dataclass, field
|
||||||
|
|
||||||
from lerobot import (
|
|
||||||
policies, # noqa: F401
|
|
||||||
)
|
|
||||||
from lerobot.datasets.transforms import ImageTransformsConfig
|
from lerobot.datasets.transforms import ImageTransformsConfig
|
||||||
from lerobot.datasets.video_utils import get_safe_default_codec
|
from lerobot.datasets.video_utils import get_safe_default_codec
|
||||||
|
|
||||||
|
|||||||
@@ -23,7 +23,7 @@ DEFAULT_INFERENCE_LATENCY = 1 / DEFAULT_FPS
|
|||||||
DEFAULT_OBS_QUEUE_TIMEOUT = 2
|
DEFAULT_OBS_QUEUE_TIMEOUT = 2
|
||||||
|
|
||||||
# All action chunking policies
|
# All action chunking policies
|
||||||
SUPPORTED_POLICIES = ["act", "smolvla", "diffusion", "pi0", "tdmpc", "vqbet"]
|
SUPPORTED_POLICIES = ["act", "smolvla", "diffusion", "pi0", "tdmpc", "vqbet", "openpi_pi0", "openpi_pi05"]
|
||||||
|
|
||||||
# TODO: Add all other robots
|
# TODO: Add all other robots
|
||||||
SUPPORTED_ROBOTS = ["so100_follower", "so101_follower"]
|
SUPPORTED_ROBOTS = ["so100_follower", "so101_follower"]
|
||||||
|
|||||||
Reference in New Issue
Block a user