Matt f04257fdbc Add test to ensure models can take int64 inputs (#17210) ...

* Add test to ensure models can take int64 inputs

* is_integer is an attribute, not a method

* Fix test when some inputs aren't tensors

* Add casts to blenderbot and blenderbot-small

* Add casts to the other failing models

2022-05-12 16:09:25 +01:00

..

benchmark

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

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

deepspeed

missing file (#17164)

2022-05-10 10:19:50 -07:00

extended

Fix issue probably-meant-fstring found at https://codereview.doctor (#16913)

2022-04-25 15:15:00 -04: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

TF: XLA bad words logits processor and list of processors (#16974)

2022-04-29 15:54:58 +01:00

models

Add OPT (#17088)

2022-05-12 12:24:35 +02:00

onnx

add mobilebert onnx configs (#17029)

2022-05-09 10:36:53 -04:00

optimization

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

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

pipelines

LogSumExp trick question_answering pipeline. (#17143)

2022-05-10 10:03:55 +02:00

sagemaker

Adapt import to new structure

2022-03-24 14:40:05 -04:00

tokenization

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

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

trainer

Ensure tensors are at least 1d for pad and concat (#17179)

2022-05-11 13:19:08 -04:00

utils

Add MLFLOW_FLATTEN_PARAMS support in MLflowCallback (#17148)

2022-05-10 14:29:18 -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

Make sure telemetry arguments are not returned as unused kwargs (#17063)

2022-05-04 07:47:57 -04:00

test_feature_extraction_common.py

Move test model folders (#17034)

2022-05-03 14:42:02 +02:00

test_modeling_common.py

ViT and Swin symbolic tracing with torch.fx (#17182)

2022-05-12 10:42:27 +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

Add test to ensure models can take int64 inputs (#17210)

2022-05-12 16:09:25 +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 #16660 (tokenizers setters of ids of special tokens) (#16661)

2022-04-13 07:49:06 -04:00