amyeroberts 9fc34235fa Use shape_list to safely get shapes for Swin (#17591) ...

* Use shape_list to safely get shapes

* Add relevant test

* Tidy and add metrics

* Resolve dynamic shaping issues and move test

* Tidy up and all samples in batch

* Formatting

2022-06-09 15:50:50 +02:00

..

benchmark

[Test refactor 1/5] Per-folder tests reorganization (#15725)

2022-02-23 15:46:28 -05:00

deepspeed

BLOOM (#17474)

2022-06-09 12:00:40 +02: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

[Generation] Fix Transition probs (#17311)

2022-05-19 22:17:02 +02:00

models

BLOOM (#17474)

2022-06-09 12:00:40 +02:00

onnx

Add ONNX support for ConvNeXT (#17627)

2022-06-09 09:31:02 -04:00

optimization

[Test refactor 1/5] Per-folder tests reorganization (#15725)

2022-02-23 15:46:28 -05:00

pipelines

Adding batch_size test to QA pipeline. (#17330)

2022-05-19 14:28:12 -04: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

Extend Transformers Trainer Class to Enable CPU AMP and Integrate Intel Extension for PyTorch (#17138)

2022-06-08 09:41:57 -04:00

utils

CLI: tool to convert PT into TF weights and open hub PR (#17497)

2022-06-01 18:52:07 +01: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

Black preview (#17217)

2022-05-12 16:25:55 -04:00

test_feature_extraction_common.py

[Json configs] Make json prettier for all saved tokenizer files & ensure same json format for all processors (tok + feat_extract) (#17457)

2022-05-31 17:07:30 +02:00

test_modeling_common.py

has_attentions - consistent test skipping logic and tf tests (#17495)

2022-06-09 09:50:03 +02:00

test_modeling_flax_common.py

[Flax] improve large model init and loading (#16148)

2022-04-19 14:19:55 +02:00

test_modeling_tf_common.py

Use shape_list to safely get shapes for Swin (#17591)

2022-06-09 15:50:50 +02: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

[Json configs] Make json prettier for all saved tokenizer files & ensure same json format for all processors (tok + feat_extract) (#17457)

2022-05-31 17:07:30 +02:00