Commit Graph

5 Commits

Author SHA1 Message Date
nemo 54c25a4400 Disallow PEFT training on non-pretrained policies
At first I thought it would make sense to have this feature
in case you want to fine-tune a pre-trained section but in the
end it makes more trouble than it's worth.

It's still possible to allow this in the future when a concrete
need arises.
2025-12-19 11:50:29 +01:00
nemo 51876e7f55 Add test checking that PEFT actually reduces params 2025-12-16 17:58:27 +01:00
nemo e544562c39 formatting 2025-11-24 18:52:18 +01:00
nemo 83aac1b42e Make sure push_to_hub works
Since PEFT only wraps `push_to_hub` and not `push_model_to_hub`, the reference
to `self` in `policy.push_model_to_hub` is the unwrapped policy which, of course,
doesn't know anything about PEFT.

To make the upload process aware of PEFT, we pass the unwrapped policy down to
`push_model_to_hub` as a kwarg. This is not ideal but I think it is the best way
for now.
2025-11-24 18:50:02 +01:00
nemo 7f42338ae9 Add CLI end-to-end tests
Currently there don't seem to be any way to test the CLI commands.
Since this change mostly happens in those I thought it best to add
a way to test these commands end-to-end.

More integrated commands like `lerobot-record` need patching but
standalone commands like training seem to work fine.
2025-10-16 16:08:20 +02:00