Commit Graph

6 Commits

Author SHA1 Message Date
Maximellerbach e9c171ead0 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-18 12:55:55 +02:00
Maximellerbach 1f80d91607 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-15 14:34:41 +02:00
Maximellerbach ddaff399b5 adding more tests to ensure good coverage 2026-05-13 15:55:04 +02:00
Maximellerbach 96a8efddf0 adding guards to avoid needing transformers and diffusers for type checking and basic tests 2026-05-12 11:01:41 +02:00
Maximellerbach 2e9ba42e1b linting 2026-05-12 11:01:41 +02:00
ginwind 3144029814 (feat)policies: add VLA-JEPA 2026-05-12 11:01:41 +02:00