diff --git a/src/fairseq2/models/jepa/loader.py b/src/fairseq2/models/jepa/loader.py index a4e546509..a38c613e7 100644 --- a/src/fairseq2/models/jepa/loader.py +++ b/src/fairseq2/models/jepa/loader.py @@ -41,7 +41,6 @@ def convert_jepa_checkpoint( def convert_jepa_encoder_checkpoint( checkpoint: dict[str, Any], config: JepaConfig ) -> dict[str, Any]: - del checkpoint["module.backbone.pos_embed"] new_checkpoint = {}