add assertions and docs

support multiple task ids

feat: assert return_dict

fix: same assertions in other models

fix: assert is None for other kwargs too

feat: added head_mask

added classifier dropout

fix: formatting

fix: formatting

feat: added further GLUE models

feat: added BertForSequenceClassification

fix: cast mask to bool

reference the flash attention GitHub

fix: move flash components into top-level

feat: try to fix import error

feat: moved flash attention code into this repository

add tokenizer

feat: added encode method

fix: try to skip initialization of task type embeddings

fix: try to skip initialization of task type embeddings

feat: added option for QK normalization

fix: removed obscure config options

feat: added small config

feat: implement task type embeddings (#1)

feat: added back option not to use flash attention

feat: support gradient checkpointing

Added additional config options

removed unused imports

removed __init__ from BertPretrainedModel

added config_class and base_model_prefix

Fixed typo

Fixed typo

Try to subclass PretrainedModel

Try to subclass PretrainedModel

strict=True for debugging

try to simplify checkpointing

changed model_type

feat: added dense_seq_output to config

removed debugging

debugging

debugging

debugging assertion

fix: fixed get_input_embeddings method

feat: added get_input_embeddings method to BertForPreTraining

feat: fixed _from_config

feat: changed model_type

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