WIP Add save_policy_to_safetensors
This commit is contained in:
parent
d84e37803f
commit
6e15e9664b
|
@ -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)
|
Loading…
Reference in New Issue