Fill-Mask
Safetensors
new
custom_code

feat(transformers): Add native Transformers config

#2

What does this PR do?

β†’ Sets model_type from new to gte and 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 3.0e-06), and the checkpoint is covered by the parity harness in the Transformers PR.

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

Sign up or log in to comment