Update README.md
Browse files
README.md
CHANGED
|
@@ -167,7 +167,7 @@ pre-trained checkpoints from HuggingFace.
|
|
| 167 |
local_dir = "pretrained_ckpts/" + model_name
|
| 168 |
os.makedirs(local_dir, exist_ok=True)
|
| 169 |
print(f"downloading {model_name} to {local_dir}...")
|
| 170 |
-
snapshot_download(repo_id=hf_repo,
|
| 171 |
```
|
| 172 |
|
| 173 |
- Under the ech checkpoint directory `pretrained_ckpts/<model-name>`, we provide the encoder,
|
|
|
|
| 167 |
local_dir = "pretrained_ckpts/" + model_name
|
| 168 |
os.makedirs(local_dir, exist_ok=True)
|
| 169 |
print(f"downloading {model_name} to {local_dir}...")
|
| 170 |
+
snapshot_download(repo_id=hf_repo, local_dir=local_dir)
|
| 171 |
```
|
| 172 |
|
| 173 |
- Under the ech checkpoint directory `pretrained_ckpts/<model-name>`, we provide the encoder,
|