feat: add base model to the model card (#459)

2026-09-30 07:51:27 -07:00 · 2026-09-30 19:22:57 +05:30
parent 3521f8648a
commit 71e6d5eb38
1 changed files with 6 additions and 0 deletions
@@ -1272,6 +1272,12 @@ def run():
                                card.data.tags.append("abliterated")
                                if reproducibility_information != "none":
                                    card.data.tags.append("reproducible")
+
+                                # Must be a Hugging Face Hub repository ID,
+                                # so local paths are excluded.
+                                if is_hf_path(settings.model):
+                                    card.data.base_model = settings.model
+
                                card.text = (
                                    get_readme_intro(
                                        settings,