Susnato Dhar 1ac2463dfe [FA2] Add flash attention for for DistilBert (#26489) ...

* flash attention added for DistilBert

* fixes

* removed padding_masks

* Update modeling_distilbert.py

* Update test_modeling_distilbert.py

* style fix

2023-11-03 16:07:54 +00:00

..

__init__.py

Move test model folders (#17034)

2022-05-03 14:42:02 +02:00

test_modeling_distilbert.py

[FA2] Add flash attention for for DistilBert (#26489)

2023-11-03 16:07:54 +00:00

test_modeling_flax_distilbert.py

CI with num_hidden_layers=2 🚀🚀🚀 (#25266)

2023-08-02 20:22:36 +02:00

test_modeling_tf_distilbert.py

Speed up TF tests by reducing hidden layer counts (#24595)

2023-06-30 16:30:33 +01:00

test_tokenization_distilbert.py

Update quality tooling for formatting (#21480)

2023-02-06 18:10:56 -05:00