transformers
cd9274d0 - [FlaxBert] Add ForCausalLM (#16995)

Commit
3 years ago
[FlaxBert] Add ForCausalLM (#16995) * [FlaxBert] Add ForCausalLM * make style * fix output attentions * Add RobertaForCausalLM * remove comment * fix fx-to-pt model loading * remove comment * add modeling tests * add enc-dec model tests * add big_bird * add electra * make style * make repo-consitency * add to docs * remove roberta test * quality * amend cookiecutter * fix attention_mask bug in flax bert model tester * tighten pt-fx thresholds to 1e-5 * add 'copied from' statements * amend 'copied from' statements * amend 'copied from' statements * quality
Parents
  • docs/source/en/model_doc
    • File
      bert.mdx
    • File
      big_bird.mdx
    • File
      electra.mdx
    • File
      roberta.mdx
  • src/transformers
    • File
      __init__.py
    • File
      modeling_flax_outputs.py
    • models
      • auto
        • File
          modeling_flax_auto.py
      • bert
        • File
          __init__.py
        • File
          modeling_flax_bert.py
      • big_bird
        • File
          __init__.py
        • File
          modeling_flax_big_bird.py
      • electra
        • File
          __init__.py
        • File
          modeling_flax_electra.py
      • roberta
        • File
          __init__.py
        • File
          modeling_flax_roberta.py
    • utils
      • File
        dummy_flax_objects.py
  • templates/adding_a_new_model/cookiecutter-template-{{cookiecutter.modelname}}
    • File
      modeling_flax_{{cookiecutter.lowercase_modelname}}.py
  • tests
    • bert
      • File
        test_modeling_flax_bert.py
    • big_bird
      • File
        test_modeling_flax_big_bird.py
    • electra
      • File
        test_modeling_flax_electra.py
    • encoder_decoder
      • File
        test_modeling_flax_encoder_decoder.py
    • roberta
      • File
        test_modeling_flax_roberta.py
    • speech_encoder_decoder
      • File
        test_modeling_flax_speech_encoder_decoder.py
  • utils
    • File
      check_repo.py