transformers/tests
Arthur bce36ee065
Load sharded pt to flax (#18419)
* initial commit

* add small test

* add cross pt tf flag to test

* fix quality

* style

* update test with new repo

* fix failing test

* update

* fix wrong param ordering

* style

* update based on review

* update related to recent new caching mechanism

* quality

* Update based on review

Co-authored-by: sgugger <sylvain.gugger@gmail.com>

* quality and style

* Update src/transformers/modeling_flax_utils.py
Co-authored-by: sgugger <sylvain.gugger@gmail.com>
Co-authored-by: Sylvain Gugger <35901082+sgugger@users.noreply.github.com>
2022-08-12 09:48:10 +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 Generate: deprecate default max_length (#18018) 2022-07-23 18:02:03 +01:00
mixed_int8 bitsandbytes - Linear8bitLt integration into transformers models (#17901) 2022-08-10 09:13:36 +02:00
models fix owlvit tests, update docstring examples (#18586) 2022-08-11 19:10:25 +03:00
onnx Adds CLIP to models exportable with ONNX (#18515) 2022-08-10 15:47:31 -04:00
optimization [Test refactor 1/5] Per-folder tests reorganization (#15725) 2022-02-23 15:46:28 -05:00
pipelines Use commit hash to look in cache instead of calling head (#18534) 2022-08-10 11:55:18 -04:00
sagemaker transformers-cli login => huggingface-cli login (#18490) 2022-08-06 09:42:55 +02:00
tokenization fix train_new_from_iterator in the case of byte-level tokenizers (#17549) 2022-06-08 15:30:41 +02:00
trainer Enable torchdynamo with torch_tensorrt(fx path) (#17765) 2022-07-13 12:43:28 -04:00
utils Clean up hub (#18497) 2022-08-08 08:48:10 -04: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 Use commit hash to look in cache instead of calling head (#18534) 2022-08-10 11:55:18 -04:00
test_feature_extraction_common.py Use new huggingface_hub tools for download models (#18438) 2022-08-05 10:12:40 -04:00
test_modeling_common.py Use new huggingface_hub tools for download models (#18438) 2022-08-05 10:12:40 -04:00
test_modeling_flax_common.py Load sharded pt to flax (#18419) 2022-08-12 09:48:10 +02:00
test_modeling_tf_common.py Use new huggingface_hub tools for download models (#18438) 2022-08-05 10:12:40 -04: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 Use new huggingface_hub tools for download models (#18438) 2022-08-05 10:12:40 -04:00