Commit Graph

11 Commits

Author SHA1 Message Date
Maximellerbach c2088fd760 fixing misconception about multiview / singleview handling 2026-05-31 22:34:49 +02:00
Maximellerbach 9fe7a542e4 adding instructions for different embodiement + fixing some tests 2026-05-31 22:34:49 +02:00
Maximellerbach 1066d037d5 pre-commit cleanup 2026-05-31 22:34:48 +02:00
Maxime Ellerbach 553c217ee2 refactoring into using pre and post processor 2026-05-31 22:34:48 +02:00
Maximellerbach c6bf11b2d5 removing missleading future_action_window_size to just use chunk_size 2026-05-31 22:34:48 +02:00
Maximellerbach 23831f5cb5 make default params more aligned with paper and pretrained models
- adding possibility of freezing qwen backbone and world model
- added tests for weight loading
2026-05-31 22:34:48 +02:00
Maximellerbach 972b32fc99 update VLA-JEPA tests for arch changes and action_is_pad
- Switch conftest to use `action_model_type="DiT-test"` now that
  `action_num_heads` / `action_attention_head_dim` have been removed.
- Add action_head tests covering fully-padded loss (zero) and equivalence
  of action_is_pad=None vs all-zeros mask.
- Remove obsolete `test_native_to_lerobot_wm_only` test.
2026-05-31 22:34:48 +02:00
Maximellerbach a4f59cd5be adding more tests to ensure good coverage 2026-05-31 22:34:48 +02:00
Maximellerbach b370da1813 adding guards to avoid needing transformers and diffusers for type checking and basic tests 2026-05-31 22:34:47 +02:00
Maximellerbach a72b3bd2f4 linting 2026-05-31 22:31:42 +02:00
ginwind 3536d17cfb (feat)policies: add VLA-JEPA 2026-05-31 22:31:42 +02:00