Upload directory
Browse files
README.md
CHANGED
|
@@ -39,6 +39,7 @@ import torch
|
|
| 39 |
|
| 40 |
# helpfer function to download huggingface repo and use model
|
| 41 |
def download(repo_id, path, HF_TOKEN=None):
|
|
|
|
| 42 |
files_path = os.path.join(path, 'files.txt')
|
| 43 |
if not os.path.exists(files_path):
|
| 44 |
hf_hub_download(repo_id, 'files.txt', token=HF_TOKEN, local_dir=path, local_dir_use_symlinks=False)
|
|
|
|
| 39 |
|
| 40 |
# helpfer function to download huggingface repo and use model
|
| 41 |
def download(repo_id, path, HF_TOKEN=None):
|
| 42 |
+
os.makedirs(path, exist_ok=True)
|
| 43 |
files_path = os.path.join(path, 'files.txt')
|
| 44 |
if not os.path.exists(files_path):
|
| 45 |
hf_hub_download(repo_id, 'files.txt', token=HF_TOKEN, local_dir=path, local_dir_use_symlinks=False)
|