fix bug

2026-06-18 00:37:10 +00:00 · 2026-06-08 15:51:09 +02:00
parent 0afa94d67b
commit 85f05adcb9
1 changed files with 1 additions and 1 deletions
@@ -263,7 +263,7 @@ class PreTrainedPolicy(nn.Module, HubMixin, abc.ABC):
            tags=list(set(tags or []).union({"robotics", "lerobot", model_type})),
            model_name=model_type,
            datasets=dataset_repo_id,
-            base_model=base_model_mapping(model_type, None),
+            base_model=base_model_mapping.get(model_type),
        )

        template_card = (