coderofpears commited on
Commit
ed0dcee
·
verified ·
1 Parent(s): 8481d22

Upload folder using huggingface_hub

Browse files
Files changed (1) hide show
  1. modal_train.py +5 -4
modal_train.py CHANGED
@@ -53,11 +53,12 @@ def train_on_l4(hours: float = 20.0, ckpt_every: int = 250,
53
  import subprocess
54
  import sys
55
 
56
- # ---- pull latest code from HF ----
 
57
  os.makedirs("/root/clanker", exist_ok=True)
58
- subprocess.run(["hf", "download", CODE_REPO,
59
- "--local-dir", "/root/clanker",
60
- "--repo-type", "model"], check=True)
61
  os.chdir("/root/clanker")
62
 
63
  data_dir, ckpt_dir = "/vol/data", "/vol/checkpoints"
 
53
  import subprocess
54
  import sys
55
 
56
+ # ---- pull latest code from HF (use huggingface_hub, not the `hf` CLI) ----
57
+ from huggingface_hub import snapshot_download
58
  os.makedirs("/root/clanker", exist_ok=True)
59
+ snapshot_download(repo_id=CODE_REPO, repo_type="model",
60
+ local_dir="/root/clanker",
61
+ token=os.environ.get("HF_TOKEN"))
62
  os.chdir("/root/clanker")
63
 
64
  data_dir, ckpt_dir = "/vol/data", "/vol/checkpoints"