Update hf_merge.py

hf_merge.py

@@ -114,16 +114,6 @@ class ModelMerger:
         copy_nontensor_files(base_model_path, output_dir)
         save_tensor_map(self.tensor_map, output_dir)
 
-    def is_repo_exists(repo_id, token):
-        from huggingface_hub import HfApi
-        api = HfApi(token=token)
-        try:
-            if api.repo_exists(repo_id=repo_id): return True
-            else: return False
-        except Exception as e:
-            print(e)
-            print(f"Error: Failed to connect {repo_id}.")
-            return True # for safe
     
     
     def upload_model(self, output_dir, commit_message):
@@ -134,25 +124,21 @@ class ModelMerger:
                 output_dir (str): The path to the output directory containing the merged model.
                 commit_message (str): The commit message for the upload.
             """
-            try:
-                # Create a new repository if it doesn't exist
-                if not self.api.repo_exists(repo_id=self.repo_id, token=self.token):
-                    self.api.create_repo(repo_id=self.repo_id, token=self.token, private=True)
     
                 # Upload the folder to the repository
-                self.api.upload_folder(
-                    repo_id=self.repo_id,
-                    folder_path=output_dir,
-                    commit_message=commit_message,
-                    token=self.token
+        self.api.upload_folder(
+                repo_id=self.repo_id,
+                folder_path=output_dir,
+                commit_message=commit_message,
+                token=self.token
                 )
-                url = f"https://huggingface.co/{self.repo_id}"
-                logging.info(f"Model uploaded successfully to {url}")
-                return url
-            except Exception as e:
-                logging.error(f"Error: Failed to upload to {self.repo_id}.")
-                logging.error(e)
-                return ""
+            url = f"https://huggingface.co/{self.repo_id}"
+            logging.info(f"Model uploaded successfully to {url}")
+            return url
+        except Exception as e:
+            logging.error(f"Error: Failed to upload to {self.repo_id}.")
+            logging.error(e)
+            return ""
 def get_max_vocab_size(repo_list):
     """
     Get the maximum vocabulary size from a list of repositories.