| |
| |
| |
| |
| @@ -117,27 +117,40 @@ def push_to_hf(dir_path: str, repo_name: str, num_policies: int = 1): |
| exist_ok=True, |
| ) |
| |
| - # Upload folders |
| - folders = [".summary"] |
| - for policy_id in range(num_policies): |
| - folders.append(f"checkpoint_p{policy_id}") |
| - for f in folders: |
| - if os.path.exists(os.path.join(dir_path, f)): |
| - upload_folder( |
| - repo_id=repo_name, |
| - folder_path=os.path.join(dir_path, f), |
| - path_in_repo=f, |
| - ) |
| - |
| - # Upload files |
| - files = ["config.json", "README.md", "replay.mp4"] |
| - for f in files: |
| - if os.path.exists(os.path.join(dir_path, f)): |
| - upload_file( |
| - repo_id=repo_name, |
| - path_or_fileobj=os.path.join(dir_path, f), |
| - path_in_repo=f, |
| - ) |
| + upload_folder( |
| + repo_id=repo_name, |
| + folder_path=dir_path, |
| + path_in_repo=f, |
| + allow_patterns=[ |
| + ".summary/*", |
| + "config.json", |
| + "README.md", |
| + "replay.mp4", |
| + ] |
| + + [f"checkpoint_p{policy_id}/*" for policy_id in range(num_policies)], |
| + ) |
| + |
| + # # Upload folders |
| + # folders = [".summary"] |
| + # for policy_id in range(num_policies): |
| + # folders.append(f"checkpoint_p{policy_id}") |
| + # for f in folders: |
| + # if os.path.exists(os.path.join(dir_path, f)): |
| + # upload_folder( |
| + # repo_id=repo_name, |
| + # folder_path=os.path.join(dir_path, f), |
| + # path_in_repo=f, |
| + # ) |
| + |
| + # # Upload files |
| + # files = ["config.json", "README.md", "replay.mp4"] |
| + # for f in files: |
| + # if os.path.exists(os.path.join(dir_path, f)): |
| + # upload_file( |
| + # repo_id=repo_name, |
| + # path_or_fileobj=os.path.join(dir_path, f), |
| + # path_in_repo=f, |
| + # ) |
| |
| log.info(f"The model has been pushed to {repo_url}") |
| |
|
|
