mirror of
https://github.com/huggingface/transformers.git
synced 2025-07-31 02:02:21 +06:00
8881f38a4f
* 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> |
||
|---|---|---|
| .. | ||
| __init__.py | ||
| {{cookiecutter.lowercase_modelname}}.md | ||
| configuration_{{cookiecutter.lowercase_modelname}}.py | ||
| configuration.json | ||
| modeling_{{cookiecutter.lowercase_modelname}}.py | ||
| modeling_flax_{{cookiecutter.lowercase_modelname}}.py | ||
| modeling_tf_{{cookiecutter.lowercase_modelname}}.py | ||
| test_modeling_{{cookiecutter.lowercase_modelname}}.py | ||
| test_modeling_flax_{{cookiecutter.lowercase_modelname}}.py | ||
| test_modeling_tf_{{cookiecutter.lowercase_modelname}}.py | ||
| to_replace_{{cookiecutter.lowercase_modelname}}.py | ||
| tokenization_{{cookiecutter.lowercase_modelname}}.py | ||
| tokenization_fast_{{cookiecutter.lowercase_modelname}}.py | ||