fix: fixed get_input_embeddings method

feat: added get_input_embeddings method to BertForPreTraining

feat: fixed _from_config

removed from_config

fix: try to get from_config to work

feat: added from_config, also pass additional kwargs from config to model

feat: updated modeling_bert.py to allow MLM-only training

feat: reverted monkey patch

feat: try to monkey-patch index_first_axis

feat: try to fix compilation

feat: added debug print

feat: updated .to() override to handle kwargs

Revert "feat updated debug pring"

feat updated debug pring

fix: try to fix .to(torch.float16) with ALiBi

Revert "feat: added back option to disable flash attention"

feat: added back option to disable flash attention

fix: always use flash attention

revert

fix: fix typo

fix: try to fix AutoModel.from_config

fix: try to fix AutoModel.from_pretrained

fix: try to fix import error

feat: added top-level docstring, made it compatible with AutoModel

initial commit