| from huggingface_hub import hf_hub_download | |
| import shutil | |
| # cached_path = hf_hub_download( | |
| # repo_id="returnzeros/anyrobot_bimanual", | |
| # filename="ACT_DP_multitask" | |
| # ) | |
| # shutil.copy(cached_path, "./policy") | |
| # print(cached_path) | |
| from huggingface_hub import list_repo_files, hf_hub_download | |
| import os, shutil | |
| repo_id = "returnzeros/anyrobot_bimanual" | |
| local_root = "./anyrobot_bimanual_policy" | |
| os.makedirs(local_root, exist_ok=True) | |
| file_list = list_repo_files(repo_id) | |
| for file in file_list: | |
| print(f"? Downloading: {file}") | |
| path = hf_hub_download(repo_id=repo_id, filename=file) | |
| local_path = os.path.join(local_root, file) | |
| os.makedirs(os.path.dirname(local_path), exist_ok=True) | |
| shutil.copy(path, local_path) | |
| print(f"load root path : {local_root}") | |