Revert "Revert "fix: disable LoRA export for now" (#308)" (#319)

This reverts commit da92f745de.
This commit is contained in:
Philipp Emanuel Weidmann
2026-05-03 07:25:00 +05:30
committed by GitHub
parent 43f8e86a84
commit 216c089974
+23 -18
View File
@@ -131,26 +131,31 @@ def obtain_merge_strategy(settings: Settings, model: Model) -> str | None:
) )
print() print()
strategy = prompt_select( strategy = prompt_select(
"How do you want to proceed?", "How do you want to proceed?",
choices=[ choices=[
Choice( Choice(
title="Merge LoRA into full model" title="Merge LoRA into full model"
+ ( + (
"" ""
if settings.quantization == QuantizationMethod.NONE if settings.quantization == QuantizationMethod.NONE
else " (requires sufficient RAM)" else " (requires sufficient RAM)"
),
value="merge",
), ),
value="merge", Choice(
), title="Cancel",
Choice( value="cancel",
title="Save LoRA adapter only (can be merged later)", ),
value="adapter", ],
), )
],
)
return strategy if strategy == "cancel":
return None
return strategy
else:
return "merge"
def run(): def run():