admin/lerobot
1
0
Fork 0
mirror of https://github.com/huggingface/lerobot.git synced 2026-07-24 10:16:09 +00:00
Code Issues Packages Projects Releases Wiki Activity

fix bug

Browse Source
This commit is contained in:
Nikodem Bartnik
2026-06-08 15:51:09 +02:00
parent 0afa94d67b
commit 85f05adcb9
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files
src/lerobot/policies/pretrained.py
+1 -1
View File
@@ -263,7 +263,7 @@ class PreTrainedPolicy(nn.Module, HubMixin, abc.ABC):
tags=list(set(tags or []).union({"robotics", "lerobot", model_type})), tags=list(set(tags or []).union({"robotics", "lerobot", model_type})),
model_name=model_type, model_name=model_type,
datasets=dataset_repo_id, datasets=dataset_repo_id,
base_model=base_model_mapping(model_type, None), base_model=base_model_mapping.get(model_type),
) )
template_card = ( template_card = (