admin/lerobot
1
0
Fork 0
mirror of https://github.com/huggingface/lerobot.git synced 2026-07-13 04:51:59 +00:00
Code Issues Packages Projects Releases Wiki Activity

make tokenizer/text-encoder model ids configurable + some nits

Browse Source
This commit is contained in:
Maxime Ellerbach
2026-06-23 09:07:16 +00:00
parent 5f6ddab629
commit 343ab5f99c
5 changed files with 48 additions and 25 deletions
Download Patch File Download Diff File Expand all files Collapse all files
tests/policies/fastwam/test_fastwam_policy.py
+3 -2
View File
@@ -73,8 +73,9 @@ def test_config_validates_features_model_ids_and_saved_auto_route(tmp_path):
assert cfg.robot_state_feature.shape == (8,)
with pytest.raises(ValueError, match="image feature"):
FastWAMConfig(input_features={OBS_STATE: PolicyFeature(type=FeatureType.STATE, shape=(8,))})
with pytest.raises(ValueError, match="tokenizer_model_id"):
FastWAMConfig(tokenizer_model_id="somebody/other-tokenizer")
assert FastWAMConfig(tokenizer_model_id="somebody/other-tokenizer").tokenizer_model_id == (
"somebody/other-tokenizer"
)
def test_preprocessor_passes_images_through_and_postprocessor_toggles_actions(tmp_path):