diff --git a/src/emma_perception/commands/download_checkpoints.py b/src/emma_perception/commands/download_checkpoints.py index fffbd4b..294510f 100644 --- a/src/emma_perception/commands/download_checkpoints.py +++ b/src/emma_perception/commands/download_checkpoints.py @@ -4,8 +4,8 @@ from emma_common.hf import download_file -HF_REPO_ID = "gpantaz/emma_models" -CHECKPOINT_NAME = "vivl_finetune_arena.ckpt" +HF_REPO_ID = "emma-heriot-watt/models" +CHECKPOINT_NAME = "vinvl_finetune_arena.ckpt" def download_vinvl_checkpoint(