Fix AIMv2 checkpoint/config compatibility with custom HF code
#7
by tewth - opened
This PR makes the uploaded checkpoint consistent with the custom modeling_aimv2.py implementation shipped in the repository.
- remap tensor names in
model.safetensorsto matchAIMv2Model.state_dict() - update
config.jsonsoarchitecturesandmodel_typealign with the shipped config/model classes
Without these changes, loading via AutoModel.from_pretrained(..., trust_remote_code=True) initializes most model weights from scratch instead of loading the published checkpoint.