Upload huggingface_download.py

huggingface_download.py

@@ -0,0 +1,29 @@
+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}")