Revert "fix: disable LoRA export for now" (#308)
This reverts commit 025ab3a881.
Co-authored-by: Andrew Patrikalakis <anrp@tri.global>
This commit is contained in:
+18
-23
@@ -127,31 +127,26 @@ 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",
|
|
||||||
),
|
),
|
||||||
Choice(
|
value="merge",
|
||||||
title="Cancel",
|
),
|
||||||
value="cancel",
|
Choice(
|
||||||
),
|
title="Save LoRA adapter only (can be merged later)",
|
||||||
],
|
value="adapter",
|
||||||
)
|
),
|
||||||
|
],
|
||||||
|
)
|
||||||
|
|
||||||
if strategy == "cancel":
|
return strategy
|
||||||
return None
|
|
||||||
|
|
||||||
return strategy
|
|
||||||
else:
|
|
||||||
return "merge"
|
|
||||||
|
|
||||||
|
|
||||||
def run():
|
def run():
|
||||||
|
|||||||
Reference in New Issue
Block a user