more work

This commit is contained in:
Kyle Corbitt
2023-08-24 23:49:44 +00:00
parent 14eae45d18
commit 40638a7848
9 changed files with 2708 additions and 463 deletions

View File

@@ -5,7 +5,7 @@ from peft import PeftModel
import os
def merge(config_file: str):
def merge_lora_model(config_file: str):
config = yaml.load(open(config_file, "r"), Loader=yaml.FullLoader)
base_model = config["base_model"]