Update hf_merge.py

hf_merge.py

@@ -1,4 +1,4 @@
-import os
+\import os
 import shutil
 import argparse
 import requests
@@ -62,9 +62,10 @@ class ModelMerger:
         save_tensor_map(self.tensor_map, output_dir)
 
     def upload_model(self, output_dir, repo_name, commit_message):
-        repo = Repository(repo_id=self.repo_id, token=self.token)
-        repo.create_branch("main", "main")
-        repo.upload_folder(output_dir, repo_path=repo_name, commit_message=commit_message)
+        repo = Repository(local_dir=output_dir, clone_from=repo_name, use_auth_token=self.token)
+        repo.git_add(auto_lfs_track=True)
+        repo.git_commit(commit_message)
+        repo.git_push()
         print(f"Model uploaded to {repo_name}")
 
 def get_max_vocab_size(repo_list):
@@ -128,4 +129,4 @@ def main():
             model_merger.upload_model(output_dir, args.repo, args.commit_message)
 
 if __name__ == "__main__":
-    main()
+    main()