From da92f745ded12eadae0d0a5ddacee739f2772422 Mon Sep 17 00:00:00 2001
From: anrp <anrp@anrp.net>
Date: Sat, 2 May 2026 00:37:47 +0000
Subject: [PATCH] Revert "fix: disable LoRA export for now" (#308)

This reverts commit 025ab3a8815d8f857a4d8bfce0e7e6af73cb0736.

Co-authored-by: Andrew Patrikalakis <anrp@tri.global>
---
 src/heretic/main.py | 41 ++++++++++++++++++-----------------------
 1 file changed, 18 insertions(+), 23 deletions(-)

diff --git a/src/heretic/main.py b/src/heretic/main.py
index e25dd81..17ca370 100644
--- a/src/heretic/main.py
+++ b/src/heretic/main.py
@@ -127,31 +127,26 @@ def obtain_merge_strategy(settings: Settings, model: Model) -> str | None:
             )
         print()
 
-        strategy = prompt_select(
-            "How do you want to proceed?",
-            choices=[
-                Choice(
-                    title="Merge LoRA into full model"
-                    + (
-                        ""
-                        if settings.quantization == QuantizationMethod.NONE
-                        else " (requires sufficient RAM)"
-                    ),
-                    value="merge",
+    strategy = prompt_select(
+        "How do you want to proceed?",
+        choices=[
+            Choice(
+                title="Merge LoRA into full model"
+                + (
+                    ""
+                    if settings.quantization == QuantizationMethod.NONE
+                    else " (requires sufficient RAM)"
                 ),
-                Choice(
-                    title="Cancel",
-                    value="cancel",
-                ),
-            ],
-        )
+                value="merge",
+            ),
+            Choice(
+                title="Save LoRA adapter only (can be merged later)",
+                value="adapter",
+            ),
+        ],
+    )
 
-        if strategy == "cancel":
-            return None
-
-        return strategy
-    else:
-        return "merge"
+    return strategy
 
 
 def run():