Upload 130 files

app.py

@@ -1511,7 +1511,8 @@ def _format_multi_model_results(results: list[dict], context: dict | None = None
 
 
 @spaces.GPU(duration=300)
-def obliterate(model_choice: str, method_choice: str, hub_repo: str,
+def obliterate(model_choice: str, method_choice: str,
+               hub_auto_push: bool, hub_repo: str,
                prompt_volume_choice: str, dataset_source_choice: str,
                custom_harmful: str, custom_harmless: str,
                # Advanced params (sliders)
@@ -1544,7 +1545,14 @@ def obliterate(model_choice: str, method_choice: str, hub_repo: str,
     model_id = MODELS.get(model_choice, model_choice)
     is_preset = model_choice in MODELS
     method = METHODS.get(method_choice, "advanced")
-    push_to_hub = hub_repo.strip() if hub_repo and hub_repo.strip() else None
+    # Resolve push-to-hub: explicit repo overrides auto-naming
+    _hub_override = hub_repo.strip() if hub_repo and hub_repo.strip() else None
+    if _hub_override:
+        push_to_hub = _hub_override
+    elif hub_auto_push:
+        push_to_hub = "auto"  # resolved to {user}/{model}-OBLITERATED at push time
+    else:
+        push_to_hub = None
     prompt_volume = PROMPT_VOLUMES.get(prompt_volume_choice, 33)
 
     # Resolve "adaptive" → telemetry-recommended method for this model
@@ -1594,7 +1602,7 @@ def obliterate(model_choice: str, method_choice: str, hub_repo: str,
 
     # Early validation: Hub repo format + HF_TOKEN
     if push_to_hub:
-        if not re.match(r'^[a-zA-Z0-9_-]+/[a-zA-Z0-9_.-]+$', push_to_hub):
+        if push_to_hub != "auto" and not re.match(r'^[a-zA-Z0-9_-]+/[a-zA-Z0-9_.-]+$', push_to_hub):
             yield (
                 "**Error:** Invalid Hub repo format. Use `username/model-name`.",
                 "", gr.update(), gr.update(), gr.update(), gr.update(),
@@ -1753,7 +1761,10 @@ def obliterate(model_choice: str, method_choice: str, hub_repo: str,
     vol_label = "all" if prompt_volume == -1 else str(prompt_volume)
     log_lines.append(f"Prompt volume: {vol_label} pairs")
     if push_to_hub:
-        log_lines.append(f"Push to Hub: {push_to_hub}")
+        if push_to_hub == "auto":
+            log_lines.append("Push to Hub: auto ({you}/{model}-OBLITERATED)")
+        else:
+            log_lines.append(f"Push to Hub: {push_to_hub}")
     if quantization:
         log_lines.append(f"Quantization: {quantization} (auto-detected for GPU fit)")
     log_lines.append("")
@@ -3498,12 +3509,18 @@ with gr.Blocks(theme=THEME, css=CSS, js=_JS, title="OBLITERATUS", fill_height=Tr
                     )
 
             with gr.Row():
-                hub_repo = gr.Textbox(
-                    label="Push to Hub (optional)",
-                    placeholder="your-username/model-name-abliterated",
-                    info="HF Hub repo ID — saves locally then uploads. "
+                hub_auto_push = gr.Checkbox(
+                    label="Auto-push to Hub",
+                    value=False,
+                    info="Auto-names your model {you}/{model}-OBLITERATED and pushes to HF Hub. "
                          "Requires HF_TOKEN env var with write access.",
                 )
+                hub_repo = gr.Textbox(
+                    label="Push to Hub (optional override)",
+                    placeholder="auto-filled when checkbox is ticked, or type your own",
+                    info="Leave blank with checkbox ticked for auto-naming, "
+                         "or enter a custom repo ID (e.g. your-username/my-model).",
+                )
             hub_warning_md = gr.Markdown("")
 
             # ── Advanced Settings (auto-populated from method preset) ────
@@ -4155,7 +4172,7 @@ The winner gets auto-pushed to HuggingFace Hub.
                     tourney_hub_org = gr.Textbox(
                         label="HF Hub Org (optional)",
                         placeholder="my-org",
-                        info="Push winner to hub-org/model-name-obliterated",
+                        info="Push winner to hub-org/model-name-OBLITERATED",
                     )
                 with gr.Column(scale=1):
                     tourney_hub_repo = gr.Textbox(
@@ -4206,7 +4223,7 @@ The winner gets auto-pushed to HuggingFace Hub.
                 # Resolve display label → HuggingFace model ID
                 model_id = model_choice.strip()
                 if model_id in MODELS:
-                    model_id = MODELS[model_id]["id"]
+                    model_id = MODELS[model_id]
 
                 hub_org_val = hub_org.strip() if hub_org and hub_org.strip() else None
                 hub_repo_val = hub_repo.strip() if hub_repo and hub_repo.strip() else None
@@ -4582,7 +4599,7 @@ Built on the shoulders of:
     # may not fire after generator teardown.
     obliterate_btn.click(
         fn=obliterate,
-        inputs=[model_dd, method_dd, hub_repo, prompt_vol_dd, dataset_dd,
+        inputs=[model_dd, method_dd, hub_auto_push, hub_repo, prompt_vol_dd, dataset_dd,
                 custom_harmful_tb, custom_harmless_tb] + _adv_controls,
         outputs=[status_md, log_box, chat_status, session_model_dd, metrics_md, ab_session_model_dd],
     ).then(

obliteratus/abliterate.py

@@ -504,6 +504,31 @@ class StageResult:
     details: dict[str, Any] = field(default_factory=dict)
 
 
+def auto_hub_repo_id(model_name: str, *, api=None) -> str:
+    """Generate a Hub repo ID like ``{hf_user}/{short_model}-OBLITERATED``.
+
+    Resolves the authenticated HF username via the API, and extracts a
+    short model name (e.g. ``meta-llama/Llama-3-8B`` → ``Llama-3-8B``).
+    """
+    import re
+
+    if api is None:
+        from huggingface_hub import HfApi
+        api = HfApi()
+
+    # Resolve HF username from token
+    user_info = api.whoami()
+    username = user_info.get("name") or user_info.get("user", "unknown")
+
+    # Extract short model name (part after '/')
+    short = model_name.split("/")[-1] if "/" in model_name else model_name
+    # Sanitize: keep alphanumeric, hyphens, dots
+    short = re.sub(r"[^a-zA-Z0-9\-.]", "-", short)
+    short = re.sub(r"-+", "-", short).strip("-")
+
+    return f"{username}/{short}-OBLITERATED"
+
+
 # ── Main pipeline ───────────────────────────────────────────────────────
 
 class AbliterationPipeline:
@@ -5921,11 +5946,17 @@ class AbliterationPipeline:
 
         # 5. Optionally push the saved directory to the Hub.
         if self.push_to_hub:
-            repo_id = self.push_to_hub
-            self.log(f"Uploading to Hub: {repo_id}")
             from huggingface_hub import HfApi
 
             api = HfApi()
+
+            # Resolve "auto" → {hf_user}/{short_model}-OBLITERATED
+            if self.push_to_hub == "auto":
+                repo_id = auto_hub_repo_id(self.model_name, api=api)
+                self.log(f"Auto-named Hub repo: {repo_id}")
+            else:
+                repo_id = self.push_to_hub
+            self.log(f"Uploading to Hub: {repo_id}")
             api.create_repo(repo_id, exist_ok=True)
             api.upload_folder(
                 folder_path=str(self.output_dir),

obliteratus/tourney.py

@@ -557,7 +557,7 @@ class TourneyRunner:
             if self.hub_repo:
                 repo_id = self.hub_repo
             else:
-                repo_id = f"{self.hub_org}/{short_model}-obliterated"
+                repo_id = f"{self.hub_org}/{short_model}-OBLITERATED"
             result.hub_repo = repo_id
 
             self.log(f"\nPushing winner to Hub: {repo_id}")