HuggingFace_transformer

Files

Dong-Yong Lee 8881f38a4f Fix beam search when using model parallel (#24969 )

* Fix GPTNeoX beam search when using parallelize

* Fix beam search idx device when using model parallel

* remove onnx related stuff

Co-authored-by: Arthur <48595927+ArthurZucker@users.noreply.github.com>

* fix: move test_beam_search_on_multi_gpu to GenerationTesterMixin

* fix: add right item to _no_split_modules of MegaPreTrainedModel

* fix: add num_beams within parallelized beam_search test

Co-authored-by: amyeroberts <22614925+amyeroberts@users.noreply.github.com>

---------

Co-authored-by: Arthur <48595927+ArthurZucker@users.noreply.github.com>
Co-authored-by: amyeroberts <22614925+amyeroberts@users.noreply.github.com>

2023-09-14 11:00:52 -04:00

__init__.py

Cleanup quality (#21493 )

2023-02-07 12:27:31 -05:00

{{cookiecutter.lowercase_modelname}}.md

Move template doc file to md (#25004 )

2023-07-21 16:49:44 -04:00

configuration_{{cookiecutter.lowercase_modelname}}.py

Fix some docs what layerdrop does (#23691 )

2023-05-23 14:50:40 -04:00

configuration.json

Add template for adding flax models (#12441 )