transformers/tests
rooa d6b6fb9963
Add CodeGen model (#17443)
* Add CodeGen model

* Add missing key and switch order of super()

* Fix torch.ones init with uint8 instead of bool

* Address comments: copy statements and doc

* update tests

* remove old model parallel

* fix batch gen tests

* fix batch gen test

* update test_gpt2_sample_max_time

* fix codgen test and revert gpt2 test change

* Fix incorrect tie_word_embedding value, typo, URL

* Fix model order in README and styling

* Reorder model list alphabetically

* Set tie_word_embedding to False by default

* Apply suggestions from code review

* Better attn mask name & remove attn masked_bias

* add tokenizer for codegen

* quality

* doc tokenizer

* fix-copies

* add CodeGenTokenizer in converter

* make truncation optional

* add test for truncation

* add copyright

* fix-copies

* fix fast tokenizer decode

* Update src/transformers/models/codegen/tokenization_codegen.py

Co-authored-by: Patrick von Platen <patrick.v.platen@gmail.com>

* increase vocab_size in tests

Co-authored-by: patil-suraj <surajp815@gmail.com>
Co-authored-by: Patrick von Platen <patrick.v.platen@gmail.com>
2022-06-24 17:10:38 +02:00
..
benchmark [Test refactor 1/5] Per-folder tests reorganization (#15725) 2022-02-23 15:46:28 -05:00
deepspeed deprecate is_torch_bf16_available (#17738) 2022-06-20 08:40:11 -04:00
extended Update self-push workflow (#17177) 2022-05-13 16:28:00 +02:00
fixtures add a warning in SpmConverter for sentencepiece's model using the byte fallback feature (#16629) 2022-04-11 11:06:10 +02:00
generation Fix top_k_top_p_filtering having unexpected behavior (#17744) 2022-06-21 21:35:55 +02:00
models Add CodeGen model (#17443) 2022-06-24 17:10:38 +02:00
onnx add onnx support for deberta and debertav2 (#17617) 2022-06-21 11:04:15 +02:00
optimization [Test refactor 1/5] Per-folder tests reorganization (#15725) 2022-02-23 15:46:28 -05:00
pipelines Add LongT5 model (#16792) 2022-06-13 22:36:58 +02:00
sagemaker Black preview (#17217) 2022-05-12 16:25:55 -04:00
tokenization fix train_new_from_iterator in the case of byte-level tokenizers (#17549) 2022-06-08 15:30:41 +02:00
trainer Prepare transformers for v0.8.0 huggingface-hub release (#17716) 2022-06-21 11:51:18 -04:00
utils 🐛 Properly raise RepoNotFoundError when not authenticated (#17651) 2022-06-10 15:41:53 +02:00
__init__.py GPU text generation: mMoved the encoded_prompt to correct device 2020-01-06 15:11:12 +01:00
test_configuration_common.py Prepare transformers for v0.8.0 huggingface-hub release (#17716) 2022-06-21 11:51:18 -04:00
test_feature_extraction_common.py Prepare transformers for v0.8.0 huggingface-hub release (#17716) 2022-06-21 11:51:18 -04:00
test_modeling_common.py Offload fixes (#17810) 2022-06-22 12:23:07 -04:00
test_modeling_flax_common.py Flax sharded (#17760) 2022-06-22 07:04:35 +02:00
test_modeling_tf_common.py Fix broken test for models with batchnorm (#17841) 2022-06-23 15:59:53 +01:00
test_sequence_feature_extraction_common.py Some tests misusing assertTrue for comparisons fix (#16771) 2022-04-19 14:44:08 +02:00
test_tokenization_common.py Fix properties of unset special tokens in non verbose mode (#17797) 2022-06-23 14:40:13 +02:00