KORMo-Team

/

KORMo-10B-sft

like 128

Follow

KORMo 136

In the case of RotaryEmbedding, the inv_freq value is calculated in the init and reused.
In Transformers 5.2, the model is loaded using the meta device, so this calculation does not take place. Consequently, in 5.2, logic was added to the _init_weights function to restore inv_freq via an else statement. In the case of KORMo, as it uses a custom _init_weights function, this logic was not applied, resulting in the issue where the RoPE value was not used during inference.
The following changes have been made to the code:

1. Added logic to restore inv_freq in _init_weights to KORMoPreTrainedModel.
2. Added the copy_ function used in _init_weights to the top of the file.
3. We resolved an issue where the original_inv_freq key value was not registered in _buffer by cloning the self.inv_freq value, which previously returned None because it was not calculated. (RotaryEmbedding)
4. We added the compute_default_rope_parameters function, which was missing in version 5.2. (RotaryEmbedding)

Compatible with both version 4.57.1 and version 5.2.
Thank you.