Training process failed with exit code 1:
File "<frozen runpy>", line 88, in _run_code
File "/workspace/lerobot/src/lerobot/scripts/lerobot_train.py", line 536, in <module>
main()
File "/workspace/lerobot/src/lerobot/scripts/lerobot_train.py", line 532, in main
train()
File "/workspace/lerobot/src/lerobot/configs/parser.py", line 233, in wrapper_inner
response = fn(cfg, *args, **kwargs)
^^^^^^^^^^^^^^^^^^^^^^^^
File "/workspace/lerobot/src/lerobot/scripts/lerobot_train.py", line 234, in train
policy = make_policy(
^^^^^^^^^^^^
File "/workspace/lerobot/src/lerobot/policies/factory.py", line 518, in make_policy
policy = policy_cls(**kwargs)
^^^^^^^^^^^^^^^^^^^^
File "/workspace/lerobot/src/lerobot/policies/groot/modeling_groot.py", line 66, in __init__
self._groot_model = self._create_groot_model()
^^^^^^^^^^^^^^^^^^^^^^^^^^
File "/workspace/lerobot/src/lerobot/policies/groot/modeling_groot.py", line 82, in _create_groot_model
model = GR00TN15.from_pretrained(
^^^^^^^^^^^^^^^^^^^^^^^^^
File "/workspace/lerobot/src/lerobot/policies/groot/groot_n1.py", line 381, in from_pretrained
pretrained_model = super().from_pretrained(
^^^^^^^^^^^^^^^^^^^^^^^^
File "/usr/local/lib/python3.12/dist-packages/transformers/modeling_utils.py", line 277, in _wrapper
return func(*args, **kwargs)
^^^^^^^^^^^^^^^^^^^^^
File "/usr/local/lib/python3.12/dist-packages/transformers/modeling_utils.py", line 4971, in from_pretrained
model = cls(config, *model_args, **model_kwargs)
^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
File "/workspace/lerobot/src/lerobot/policies/groot/groot_n1.py", line 224, in __init__
self.backbone = EagleBackbone(**config.backbone_cfg)
^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
File "/workspace/lerobot/src/lerobot/policies/groot/groot_n1.py", line 93, in __init__
self.eagle_model = AutoModel.from_config(config, trust_remote_code=True)
^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
File "/usr/local/lib/python3.12/dist-packages/transformers/models/auto/auto_factory.py", line 444, in from_config
model_class = get_class_from_dynamic_module(class_ref, repo_id, **kwargs)
^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
File "/usr/local/lib/python3.12/dist-packages/transformers/dynamic_module_utils.py", line 604, in get_class_from_dynamic_module
final_module = get_cached_module_file(
^^^^^^^^^^^^^^^^^^^^^^^
File "/usr/local/lib/python3.12/dist-packages/transformers/dynamic_module_utils.py", line 427, in get_cached_module_file
modules_needed = check_imports(resolved_module_file)
^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
File "/usr/local/lib/python3.12/dist-packages/transformers/dynamic_module_utils.py", line 260, in check_imports
raise ImportError(
ImportError: This modeling file requires the following packages that were not found in your environment: peft. Run `pip install peft`
[GROOT] Copying vendor Eagle files to cache: /workspace/lerobot/src/lerobot/policies/groot/eagle2_hg_model -> /root/.cache/huggingface/lerobot/lerobot/eagle2hg-processor-groot-n1p5
[GROOT] Assets repo: lerobot/eagle2hg-processor-groot-n1p5
Cache dir: /root/.cache/huggingface/lerobot/lerobot/eagle2hg-processor-groot-n1p5