From a981ceb50c076391968354d279fed7c36349506a Mon Sep 17 00:00:00 2001
From: Philipp Emanuel Weidmann <pew@worldwidemann.com>
Date: Sun, 3 May 2026 07:21:36 +0530
Subject: [PATCH] Revert "Revert "fix: disable LoRA export for now" (#308)"

This reverts commit da92f745ded12eadae0d0a5ddacee739f2772422.
---
 src/heretic/main.py | 41 +++++++++++++++++++++++------------------
 1 file changed, 23 insertions(+), 18 deletions(-)

diff --git a/src/heretic/main.py b/src/heretic/main.py
index 1262c221..693c3d0a 100644
--- a/src/heretic/main.py
+++ b/src/heretic/main.py
@@ -131,26 +131,31 @@ 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)"
+        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",
                 ),
-                value="merge",
-            ),
-            Choice(
-                title="Save LoRA adapter only (can be merged later)",
-                value="adapter",
-            ),
-        ],
-    )
+                Choice(
+                    title="Cancel",
+                    value="cancel",
+                ),
+            ],
+        )
+
+        if strategy == "cancel":
+            return None
 
-    return strategy
+        return strategy
+    else:
+        return "merge"
 
 
 def run():