Update hf_merge.py

hf_merge.py

@@ -3,7 +3,7 @@ import shutil
 import argparse
 import requests
 from tqdm import tqdm
-from huggingface_hub import HfApi, hf_hub_download
+from huggingface_hub import HfApi, hf_hub_download, create_repo
 from merge import merge_folder, map_tensors_to_files, copy_nontensor_files, save_tensor_map
 import logging
 
@@ -119,10 +119,10 @@ class ModelMerger:
             commit_message (str): The commit message for the upload.
         """
         # Upload the entire folder to the HuggingFace Space
+        create_repo(repo_name, repo_type="model")
         self.api.upload_folder(
             folder_path=output_dir,
             repo_id=repo_name,
-            repo_type="model",
             commit_message=commit_message
         )
         logging.info(f"Model uploaded to {repo_name}")