mirror of
https://github.com/p-e-w/heretic.git
synced 2026-06-02 05:03:33 +02:00
+23
-18
@@ -101,26 +101,31 @@ def obtain_merge_strategy(settings: Settings) -> 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():
|
||||||
|
|||||||
Reference in New Issue
Block a user