Upload hfhub.py with huggingface_hub

hfhub.py

@@ -1,23 +1,20 @@
 import os
 from pathlib import Path
 
-from huggingface_hub import (
-    HfApi,
-    Repository,
-    create_repo,
-    hf_hub_download,
-    login,
-    snapshot_download,
-)
+from huggingface_hub import HfApi, Repository, hf_hub_download, snapshot_download
 
 repo_name = "test2"
 repo_dir = Path("hfhub")
 token = os.environ["HF_TOKEN"]
+download_checkpoint_with_name = "test_checkpoint"
 
 (Path(repo_dir) / "checkpoints").mkdir(parents=True, exist_ok=True)
 
 hf_api = HfApi(token=token)
+
+# username/repo_name
 repo_name = hf_api.get_full_repo_name(repo_name)
+
 repo_url = hf_api.create_repo(
     repo_id=repo_name, repo_type="model", exist_ok=True, private=True
 )
@@ -28,3 +25,13 @@ hf_api.upload_file(
     path_or_fileobj="hfhub.py",
     path_in_repo="hfhub.py",
 )
+
+ckpt_filepath = hf_hub_download(
+    repo_id=repo_name,
+    repo_type="model",
+    cache_dir=Path(repo_dir),
+    resume_download=True,
+    subfolder="checkpoints",
+    filename=download_checkpoint_with_name,
+)
+print(ckpt_filepath)