File tree Expand file tree Collapse file tree
Expand file tree Collapse file tree Original file line number Diff line number Diff 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
161156def run ():
You can’t perform that action at this time.
0 commit comments