From a7eb52fd3ee8f163febfa43a834610ce1c8b9283 Mon Sep 17 00:00:00 2001 From: fredithefish <83492589+fredi-python@users.noreply.github.com> Date: Thu, 11 May 2023 19:42:14 +0200 Subject: [PATCH 1/3] Update merge_peft_adapters.py --- finetune/merge_peft_adapters.py | 14 ++++++++------ 1 file changed, 8 insertions(+), 6 deletions(-) diff --git a/finetune/merge_peft_adapters.py b/finetune/merge_peft_adapters.py index 790b6fd..11b09fe 100644 --- a/finetune/merge_peft_adapters.py +++ b/finetune/merge_peft_adapters.py @@ -10,6 +10,8 @@ def get_args(): parser.add_argument("--base_model_name_or_path", type=str, default="bigcode/large-model") parser.add_argument("--peft_model_path", type=str, default="/") parser.add_argument("--push_to_hub", action="store_true", default=True) + parser.add_argument("--base_model_name_or_path", type=str, default="bigcode/large-model") + parser.add_argument("--repo_name", type=str) return parser.parse_args() @@ -30,12 +32,12 @@ def main(): if args.push_to_hub: print(f"Saving to hub ...") - model.push_to_hub(f"{args.base_model_name_or_path}-merged", use_temp_dir=False, private=True) - tokenizer.push_to_hub(f"{args.base_model_name_or_path}-merged", use_temp_dir=False, private=True) + model.push_to_hub(f"{args.repo_name}-merged", use_temp_dir=False, private=False) + tokenizer.push_to_hub(f"{args.repo_name}-merged", use_temp_dir=False, private=False) else: - model.save_pretrained(f"{args.base_model_name_or_path}-merged") - tokenizer.save_pretrained(f"{args.base_model_name_or_path}-merged") - print(f"Model saved to {args.base_model_name_or_path}-merged") + model.save_pretrained(f"{args.repo_name}-merged") + tokenizer.save_pretrained(f"{args.repo_name}-merged") + print(f"Model saved to {args.repo_name}-merged") if __name__ == "__main__" : - main() \ No newline at end of file + main() From 25fbec8b9da4172aa4e679a068e2a757f2dbf48b Mon Sep 17 00:00:00 2001 From: fredithefish <83492589+fredi-python@users.noreply.github.com> Date: Thu, 11 May 2023 20:04:58 +0200 Subject: [PATCH 2/3] Update merge_peft_adapters.py --- finetune/merge_peft_adapters.py | 1 - 1 file changed, 1 deletion(-) diff --git a/finetune/merge_peft_adapters.py b/finetune/merge_peft_adapters.py index 11b09fe..2b15c18 100644 --- a/finetune/merge_peft_adapters.py +++ b/finetune/merge_peft_adapters.py @@ -10,7 +10,6 @@ def get_args(): parser.add_argument("--base_model_name_or_path", type=str, default="bigcode/large-model") parser.add_argument("--peft_model_path", type=str, default="/") parser.add_argument("--push_to_hub", action="store_true", default=True) - parser.add_argument("--base_model_name_or_path", type=str, default="bigcode/large-model") parser.add_argument("--repo_name", type=str) return parser.parse_args() From 6206b84a51954679955f11e20e75080abe68b516 Mon Sep 17 00:00:00 2001 From: fredithefish <83492589+fredi-python@users.noreply.github.com> Date: Thu, 11 May 2023 20:36:26 +0200 Subject: [PATCH 3/3] Update merge_peft_adapters.py --- finetune/merge_peft_adapters.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/finetune/merge_peft_adapters.py b/finetune/merge_peft_adapters.py index 2b15c18..11bbadb 100644 --- a/finetune/merge_peft_adapters.py +++ b/finetune/merge_peft_adapters.py @@ -31,8 +31,8 @@ def main(): if args.push_to_hub: print(f"Saving to hub ...") - model.push_to_hub(f"{args.repo_name}-merged", use_temp_dir=False, private=False) - tokenizer.push_to_hub(f"{args.repo_name}-merged", use_temp_dir=False, private=False) + model.push_to_hub(f"{args.repo_name}-merged", use_temp_dir=True, private=False) + tokenizer.push_to_hub(f"{args.repo_name}-merged", use_temp_dir=True, private=False) else: model.save_pretrained(f"{args.repo_name}-merged") tokenizer.save_pretrained(f"{args.repo_name}-merged")