mirror of
https://github.com/huggingface/lerobot.git
synced 2026-05-21 03:30:10 +00:00
Jade Choghari
11 lines
468 B
Python
11 lines
468 B
Python
from lerobot.policies.factory import make_policy
|
|
from lerobot.policies.factory import make_policy_config
|
|
from lerobot.datasets.lerobot_dataset import LeRobotDatasetMetadata
|
|
|
|
cfg = make_policy_config("xvla")
|
|
|
|
dataset_id = "lerobot/svla_so101_pickplace"
|
|
# This only downloads the metadata for the dataset, ~10s of MB even for large-scale datasets
|
|
dataset_metadata = LeRobotDatasetMetadata(dataset_id)
|
|
policy = make_policy(cfg=cfg, ds_meta=dataset_metadata)
|
|
print(policy) |