NielsRogge 7e7f743481 Add SegFormer ONNX support (#18006) ...

* Add ONNX support

* Make height and width dynamic axes

Co-authored-by: Niels Rogge <nielsrogge@Nielss-MacBook-Pro.local>

2022-08-31 20:58:44 +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: add missing **model_kwargs in sample tests (#18696)

2022-08-19 16:14:27 +01:00

mixed_int8

[bnb] Minor modifications (#18631)

2022-08-17 00:48:10 +02:00

models

[DETR] Add num_channels attribute (#18714)

2022-08-31 18:04:42 +02:00

onnx

Add SegFormer ONNX support (#18006)

2022-08-31 20:58:44 +02: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

small change (#18584)

2022-08-12 20:04:38 +02:00

utils

Add an option to HfArgumentParser.parse_{dict,json_file} to raise an Exception when there extra keys (#18692)

2022-08-31 20:26:45 +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

Fix mock in test_cached_files_are_used_when_internet_is_down (#18804)

2022-08-29 15:56:08 +02:00

test_feature_extraction_common.py

Fix mock in test_cached_files_are_used_when_internet_is_down (#18804)

2022-08-29 15:56:08 +02:00

test_modeling_common.py

Fix mock in test_cached_files_are_used_when_internet_is_down (#18804)

2022-08-29 15:56:08 +02: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

Fix mock in test_cached_files_are_used_when_internet_is_down (#18804)

2022-08-29 15:56:08 +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

Fix mock in test_cached_files_are_used_when_internet_is_down (#18804)

2022-08-29 15:56:08 +02:00