simplify the upload instructions.

README.md

@@ -172,35 +172,17 @@ Once your model has been fine-tuned, you can export it to a HuggingFace model.
 
 In order to use your model in `app.py`, you will need to export it to HuggingFace.
 
-**NOTE**: In order to export, you will need a [huggingface account](https://huggingface.co/). 
-
-<!-- You need to fork the [vampnet models repo](https://huggingface.co/hugggof/vampnet) which stores the default vampnet models.  -->
-
-Now, create a repo on huggingface. You can do this by going to the huggingface website and clicking on the "Create a new model" button.
-Copy your repo name. It should look something like `<USERNAME>/vampnet`.
-
-Now, navigate to `models/vampnet` and add your new repo as a remote:
-
-```bash
-cd ./models/vampnet && git init && git remote add origin https://huggingface.co/<YOUR_REPO_NAME> && git pull origin main
-```
-go back
-
-```bash
-cd ../../
-```
-
-Now, replace the contents of the file named `./DEFAULT_HF_MODEL_REPO` in the root folder with the name of your repo (usually `<USERNAME>/vampnet`). 
+**NOTE**: In order to export, you will need a [huggingface account](https://huggingface.co/).
 
 Now, log in to huggingface using the command line:
 ```bash
 huggingface-cli login
 ```
 
-Now, run the following command to export your model:
+Now, run the following command to export your model (replace `<your_finetuned_model_name>` with the name of your model, and `<HUGGINGFACE_USERNAME>` with your huggingface username):
 
 ```bash
-python scripts/exp/export.py --name <your_finetuned_model_name> --model latest
+python scripts/exp/export.py --name <your_finetuned_model_name> --model latest --repo <HUGGINGFACE_USERNAME>/vampnet
 ```
 
 Once that's done, your model should appear on the list of available models in the gradio interface.

scripts/exp/export.py

@@ -3,6 +3,8 @@ from pathlib import Path
 import shutil
 import argparse
 from vampnet import DEFAULT_HF_MODEL_REPO
+from huggingface_hub import create_repo, repo_exists, HfApi
+
 
 
 parser = argparse.ArgumentParser(description="Export the fine-tuned model to the repo")
@@ -14,14 +16,31 @@ parser.add_argument(
     "--model", type=str, default="latest",
     help="model version to export. check runs/<name> for available versions"
 )
+parser.add_argument(
+    "--repo", type=str, default=DEFAULT_HF_MODEL_REPO,
+    help="name of the repo to export to"
+)
 
 args = parser.parse_args()
 name = args.name
 version = args.model
 
+##
+print(f"~~~~~~~~~~~ vampnet export! ~~~~~~~~~~~~")
+print(f"exporting {name} version {version} to {args.repo}\n")
+
 run_dir = Path(f"runs/{name}")
 repo_dir = Path("models/vampnet")
 
+# create our repo
+new_repo = False
+if not repo_exists(args.repo):
+    print(f"repo {args.repo} does not exist, creating it")
+    print(f"creating a repo at {args.repo}")
+    create_repo(args.repo)
+    new_repo = True
+
+paths = []
 for part in ("coarse", "c2f"):
     outdir = repo_dir / "loras" / name 
     outdir.mkdir(parents=True, exist_ok=True)
@@ -29,13 +48,27 @@ for part in ("coarse", "c2f"):
     path = run_dir / part / version / "vampnet" / "weights.pth"
     # path.rename(outpath)
     shutil.copy(path, outpath)
+    paths.append(outpath)
     print(f"moved {path} to {outpath}")
 
-from huggingface_hub import Repository
-repo = Repository(str(repo_dir),clone_from=f"https://huggingface.co/{DEFAULT_HF_MODEL_REPO}")
-print(f"pushing {repo_dir} to {name}")
-print(f"howdy! I am pushing to {DEFAULT_HF_MODEL_REPO} :)")
-repo.push_to_hub(
-    commit_message=f"add {name}", 
-)
+print(f"uploading files to {args.repo}")
+# upload files to the repo
+
+# if it's a new repo, let's add the default models too
+defaults = [repo_dir / "c2f.pth", repo_dir / "coarse.pth", repo_dir / "codec.pth", repo_dir / "wavebeat.pth"]
+
+api = HfApi()
+
+for path in paths:
+    path_in_repo = str(path.relative_to(repo_dir))
+    print(f"uploading {path} to {args.repo}/{path_in_repo}")
+    api.upload_file(
+        path_or_fileobj=path,
+        path_in_repo=path_in_repo,
+        repo_id=args.repo,
+        token=True,
+        commit_message=f"uploading {path_in_repo}",
+    )
+
+
 print("done!!! >::0")