From 4865f9f33f30d50a9322a7bd89b5991da3de03c6 Mon Sep 17 00:00:00 2001
From: Andrew Patrikalakis <anrp@tri.global>
Date: Wed, 22 Apr 2026 08:56:36 -0400
Subject: [PATCH] Revert "fix: disable LoRA export for now"

This reverts commit 025ab3a8815d8f857a4d8bfce0e7e6af73cb0736.
---
 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 e25dd813..17ca370f 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",
-                ),
-                Choice(
-                    title="Cancel",
-                    value="cancel",
+    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)"
                 ),
-            ],
-        )
-
-        if strategy == "cancel":
-            return None
+                value="merge",
+            ),
+            Choice(
+                title="Save LoRA adapter only (can be merged later)",
+                value="adapter",
+            ),
+        ],
+    )
 
-        return strategy
-    else:
-        return "merge"
+    return strategy
 
 
 def run():