From b6353964ba77f29fea03eaa1b7b2c12a2c6a8ccf Mon Sep 17 00:00:00 2001 From: Alexander Soare Date: Wed, 27 Mar 2024 13:08:47 +0000 Subject: [PATCH] fix bug: use provided revision instead of hardcoded one --- lerobot/scripts/eval.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/lerobot/scripts/eval.py b/lerobot/scripts/eval.py index 1de0bb0e..28a25e43 100644 --- a/lerobot/scripts/eval.py +++ b/lerobot/scripts/eval.py @@ -268,7 +268,7 @@ if __name__ == "__main__": # TODO(alexander-soare): Save and load stats in trained model directory. stats_path = None elif args.hub_id is not None: - folder = Path(snapshot_download(args.hub_id, revision="v1.0")) + folder = Path(snapshot_download(args.hub_id, revision=args.revision)) cfg = hydra.initialize(config_path=str(_relative_path_between(folder, Path(__file__).parent))) cfg = hydra.compose("config", args.overrides) cfg.policy.pretrained_model_path = folder / "model.pt"