Rename Diffusion Policy model repository (#136)

This commit is contained in:
Alexander Soare
2024-05-05 12:06:58 +01:00
committed by GitHub
parent f3bba0270d
commit f6fe162432
4 changed files with 7 additions and 7 deletions

View File

@@ -58,7 +58,7 @@ def test_examples_3_and_2():
file_contents = _find_and_replace(
file_contents,
[
('pretrained_policy_name = "lerobot/diffusion_policy_pusht_image"', ""),
('pretrained_policy_name = "lerobot/diffusion_pusht"', ""),
("pretrained_policy_path = Path(snapshot_download(pretrained_policy_name))", ""),
(
'# pretrained_policy_path = Path("outputs/train/example_pusht_diffusion")',