diff --git a/finetune/merge_peft_adapters.py b/finetune/merge_peft_adapters.py
index 790b6fd..11bbadb 100644
--- a/finetune/merge_peft_adapters.py
+++ b/finetune/merge_peft_adapters.py
@@ -10,6 +10,7 @@ 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("--repo_name", type=str)
 
     return parser.parse_args()
 
@@ -30,12 +31,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=True, private=False)
+        tokenizer.push_to_hub(f"{args.repo_name}-merged", use_temp_dir=True, 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()