Addition of HF compatible checkpoints
#2
by
yaswanthgali
- opened
First of all great work and thanks for these strong models. I'm working on the PR to add native support for AimV2 in Transformers library. It would be helpful to transfer the HF-compatible checkpoints to the official repo to make user's life easier. I've converted four checkpoints covering different types for starters.
PR: https://github.com/huggingface/transformers/pull/36625
- yaswanthgali/aimv2-large-patch14-336-HF
- yaswanthgali/aimv2-large-patch14-224-HF
- yaswanthgali/aimv2-large-patch14-224-lit-HF
- yaswanthgali/aimv2-large-patch14-native-HF
LMK if you want to update the existing checkpoints or a separate set of checkpoints for HF compatible ones which IMO is not required as previously we use to load current ones with trust_remote_code=True arg .