From bc077540a0d369a39339a97500d00ee08eaa91a1 Mon Sep 17 00:00:00 2001
From: Ayush Vishwakarma <officialayush5839@gmail.com>
Date: Tue, 20 Oct 2026 22:12:06 +0530
Subject: [PATCH] feat: make upload tags configurable (#506)

---
 config.default.toml   |  7 +++++++
 config.nohumor.toml   |  4 ++++
 config.noslop.toml    |  4 ++++
 src/heretic/config.py |  9 +++++++++
 src/heretic/main.py   | 14 +++++++++-----
 tests/test_config.py  | 25 +++++++++++++++++++++++++
 6 files changed, 58 insertions(+), 5 deletions(-)

diff --git a/config.default.toml b/config.default.toml
index f8d13da..0c21bf2 100644
--- a/config.default.toml
+++ b/config.default.toml
@@ -99,6 +99,13 @@ study_checkpoint_dir = "checkpoints"
 # Maximum size for individual safetensors files generated when exporting a model.
 max_shard_size = "5GB"
 
+# Tags to add to the Hugging Face repository when uploading the model.
+upload_tags = [
+    "uncensored",
+    "decensored",
+    "abliterated",
+]
+
 # System prompt to use when prompting the model.
 system_prompt = "You are a helpful assistant."
 
diff --git a/config.nohumor.toml b/config.nohumor.toml
index 38f080e..56c4886 100644
--- a/config.nohumor.toml
+++ b/config.nohumor.toml
@@ -3,6 +3,10 @@
 
 max_response_length = 300
 
+upload_tags = [
+    "no-humor",
+]
+
 [scorer.KeywordRate]
 score_name = "Responses with humor"
 
diff --git a/config.noslop.toml b/config.noslop.toml
index f9eed07..99bd09a 100644
--- a/config.noslop.toml
+++ b/config.noslop.toml
@@ -3,6 +3,10 @@
 
 max_response_length = 300
 
+upload_tags = [
+    "no-slop",
+]
+
 system_prompt = "You are a professional writer."
 
 [scorer.KeywordRate]
diff --git a/src/heretic/config.py b/src/heretic/config.py
index c5081a1..cee7df1 100644
--- a/src/heretic/config.py
+++ b/src/heretic/config.py
@@ -541,6 +541,15 @@ class Settings(BaseSettings):
         description='Which reproducibility information to add to the Hugging Face repository: "full", "basic", "none", or unset to prompt the user.',
     )
 
+    upload_tags: list[str] = Field(
+        default=[
+            "uncensored",
+            "decensored",
+            "abliterated",
+        ],
+        description="Tags to add to the Hugging Face repository when uploading the model.",
+    )
+
     ignore_mismatches: bool | None = Field(
         default=None,
         description="Whether to attempt to reproduce the model even if there are environment mismatches, or unset to prompt the user.",
diff --git a/src/heretic/main.py b/src/heretic/main.py
index 6162c7d..4d70c87 100644
--- a/src/heretic/main.py
+++ b/src/heretic/main.py
@@ -1172,11 +1172,15 @@ def run():
                                     card.data = ModelCardData()
                                 if card.data.tags is None:
                                     card.data.tags = []
-                                card.data.tags.append("heretic")
-                                card.data.tags.append("uncensored")
-                                card.data.tags.append("decensored")
-                                card.data.tags.append("abliterated")
-                                if reproducibility_information != "none":
+                                if "heretic" not in card.data.tags:
+                                    card.data.tags.append("heretic")
+                                for tag in settings.upload_tags:
+                                    if tag not in card.data.tags:
+                                        card.data.tags.append(tag)
+                                if (
+                                    reproducibility_information != "none"
+                                    and "reproducible" not in card.data.tags
+                                ):
                                     card.data.tags.append("reproducible")
 
                                 # Must be a Hugging Face Hub repository ID,
diff --git a/tests/test_config.py b/tests/test_config.py
index 351abae..8b440d7 100644
--- a/tests/test_config.py
+++ b/tests/test_config.py
@@ -134,5 +134,30 @@ class ModelValidateInitOnlyTests(unittest.TestCase):
         self.assertEqual(settings.max_memory, {"0": "20GiB"})
 
 
+class SettingsUploadTagsTests(unittest.TestCase):
+    def test_default_upload_tags(self) -> None:
+        with patch("sys.argv", ["heretic"]):
+            settings = Settings(model="dummy")
+            self.assertEqual(
+                settings.upload_tags,
+                ["uncensored", "decensored", "abliterated"],
+            )
+
+    def test_custom_upload_tags_override(self) -> None:
+        with patch("sys.argv", ["heretic"]):
+            settings = Settings(model="dummy", upload_tags=["custom-tag", "experiment"])
+            self.assertEqual(settings.upload_tags, ["custom-tag", "experiment"])
+
+    def test_empty_upload_tags(self) -> None:
+        with patch("sys.argv", ["heretic"]):
+            settings = Settings(model="dummy", upload_tags=[])
+            self.assertEqual(settings.upload_tags, [])
+
+    def test_upload_tags_from_cli(self) -> None:
+        with patch("sys.argv", ["heretic", "--upload-tags", '["cli-tag", "test"]']):
+            settings = Settings(model="dummy")
+            self.assertEqual(settings.upload_tags, ["cli-tag", "test"])
+
+
 if __name__ == "__main__":
     unittest.main()