Skip to content

Commit 54388dc

Browse files
committed
Revert "Revert "Revert "fix: disable LoRA export for now" (#308)" (#319)"
This reverts commit 216c089.
1 parent 40bf2fe commit 54388dc

1 file changed

Lines changed: 18 additions & 23 deletions

File tree

src/heretic/main.py

Lines changed: 18 additions & 23 deletions
Original file line numberDiff line numberDiff line change
@@ -131,31 +131,26 @@ def obtain_merge_strategy(settings: Settings, model: Model) -> str | None:
131131
)
132132
print()
133133

134-
strategy = prompt_select(
135-
"How do you want to proceed?",
136-
choices=[
137-
Choice(
138-
title="Merge LoRA into full model"
139-
+ (
140-
""
141-
if settings.quantization == QuantizationMethod.NONE
142-
else " (requires sufficient RAM)"
143-
),
144-
value="merge",
145-
),
146-
Choice(
147-
title="Cancel",
148-
value="cancel",
134+
strategy = prompt_select(
135+
"How do you want to proceed?",
136+
choices=[
137+
Choice(
138+
title="Merge LoRA into full model"
139+
+ (
140+
""
141+
if settings.quantization == QuantizationMethod.NONE
142+
else " (requires sufficient RAM)"
149143
),
150-
],
151-
)
152-
153-
if strategy == "cancel":
154-
return None
144+
value="merge",
145+
),
146+
Choice(
147+
title="Save LoRA adapter only (can be merged later)",
148+
value="adapter",
149+
),
150+
],
151+
)
155152

156-
return strategy
157-
else:
158-
return "merge"
153+
return strategy
159154

160155

161156
def run():

0 commit comments

Comments
 (0)