WIP Add save_policy_to_safetensors

This commit is contained in:
Simon Alibert 2024-04-30 19:25:11 +02:00
parent d84e37803f
commit 6e15e9664b
1 changed files with 10 additions and 0 deletions

View File

@ -0,0 +1,10 @@
import shutil
from pathlib import Path
def save_policy_to_safetensors(output_dir, repo_id="lerobot/pusht"):
...
repo_dir = Path(output_dir) / repo_id
if repo_dir.exists():
shutil.rmtree(repo_dir)