From 025ab3a8815d8f857a4d8bfce0e7e6af73cb0736 Mon Sep 17 00:00:00 2001 From: Philipp Emanuel Weidmann Date: Sat, 14 Feb 2026 16:56:12 +0530 Subject: [PATCH] fix: disable LoRA export for now Workaround for #152 --- 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 d33188f..50befc7 100644 --- a/src/heretic/main.py +++ b/src/heretic/main.py @@ -101,26 +101,31 @@ def obtain_merge_strategy(settings: Settings) -> 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", + ), + ], + ) - return strategy + if strategy == "cancel": + return None + + return strategy + else: + return "merge" def run():