Update hf_merge.py

hf_merge.py

@@ -11,7 +11,9 @@ import sys
 import yaml
 from huggingface_hub import snapshot_download
 from huggingface_hub import HfApi, hf_hub_download
+import spaces
 
+@spaces.GPU
 def parse_arguments():
     parser = argparse.ArgumentParser(description="Merge HuggingFace models")
     parser.add_argument('repo_list', type=str, help='File containing list of repositories to merge, supports mergekit yaml or txt')