mirror of
https://github.com/p-e-w/heretic.git
synced 2026-06-01 20:58:47 +02:00
+23
-18
@@ -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():
|
||||
|
||||
Reference in New Issue
Block a user