Sapiensia/LLaVA

1import argparse2from llava.model.builder import load_pretrained_model3from llava.mm_utils import get_model_name_from_path4 5 6def merge_lora(args):7    model_name = get_model_name_from_path(args.model_path)8    tokenizer, model, image_processor, context_len = load_pretrained_model(args.model_path, args.model_base, model_name, device_map='cpu')9 10    model.save_pretrained(args.save_model_path)11    tokenizer.save_pretrained(args.save_model_path)12 13 14if __name__ == "__main__":15    parser = argparse.ArgumentParser()16    parser.add_argument("--model-path", type=str, required=True)17    parser.add_argument("--model-base", type=str, required=True)18    parser.add_argument("--save-model-path", type=str, required=True)19 20    args = parser.parse_args()21 22    merge_lora(args)23