lm-finetuner: Merger: Dequantize before merging lora adapter; More lora options;

This commit is contained in:
extio-de
2024-12-01 13:03:45 +01:00
parent aa958eeb81
commit ff2385bf50
5 changed files with 21 additions and 5 deletions

View File

@@ -10,13 +10,19 @@ class Merger:
if not context.storeAdapter:
raise Exception("Configuration problem: mergeFull requires storeAdapter")
print("Reloading original model on CPU")
baseModel = ModelLoader().load(context.locBaseModel, True, False, context)
print("Reloading model")
baseModel = ModelLoader().load(context.locBaseModel, False, True, context)
peftModel = PeftModel.from_pretrained(baseModel, context.locAdapter)
print("Merging")
mergedModel = peftModel.merge_and_unload()
print("Dequantize")
try:
mergedModel = mergedModel.dequantize()
except:
pass
print("Storing model")
mergedModel.save_pretrained(save_directory = context.locFull)