feat(transformers): Add native Transformers config

#17
by harshaljanjani - opened

What does this PR do?

โ†’ Adds rope_parameters, so the checkpoint loads directly with AutoModel / AutoConfig in Transformers ๐Ÿค— without trust_remote_code.
โ†’ Transformers support is being added in https://github.com/huggingface/transformers/pull/48416
โ†’ โš ๏ธ Please hold this until the Transformers PR above has landed ๐Ÿค—

Compatibility

โ†’ rope_scaling, rope_theta and auto_map are left untouched, so nothing changes for anyone on an older version.

Produced by?

โ†’ rope_parameters is the standardised form of the existing rope_theta. This checkpoint leaves rope_scaling at null, so the standardised form is a plain default rope with the same base.
โ†’ Outputs are reproduced against the original implementation on this checkpoint (max relative difference 1.2e-06), and the checkpoint is covered by the parity harness in the Transformers PR.

harshaljanjani changed pull request status to open
harshaljanjani changed pull request title from โš ๏ธ DO NOT MERGE: feat(transformers): Add native Transformers config to feat(transformers): Add native Transformers config
Ready to merge
This branch is ready to get merged automatically.

Sign up or log in to comment