Upload folder using huggingface_hub
Browse files- 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 |
-
|
| 59 |
-
|
| 60 |
-
|
| 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"
|