Michael Benayoun 4e2f4a92dd [FX] Symbolic trace for Bloom (#18356) ...

* Bloom model can now be traced

* Bloom traced model can be torch scripted and serialized

* Bloom can be traced with variable keyword arguments

* Enable XLNet support

* Disable XLNet for now

2022-07-29 16:12:27 +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

models

[FX] Symbolic trace for Bloom (#18356)

2022-07-29 16:12:27 +02:00

onnx

add ONNX support for LeVit (#18154)

2022-07-18 15:17:07 +02:00

optimization

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

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

pipelines

Custom pipeline (#18079)

2022-07-19 12:02:35 +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

Enable torchdynamo with torch_tensorrt(fx path) (#17765)

2022-07-13 12:43:28 -04:00

utils

Update serving code to enable saved_model=True (#18153)

2022-07-22 18:05:38 +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

[From pretrained] Allow download from subfolder inside model repo (#18184)

2022-07-19 11:53:53 +02:00

test_feature_extraction_common.py

Compute min_resolution in prepare_image_inputs (#17915)

2022-06-29 10:30:20 +02:00

test_modeling_common.py

Fixes torch jit tracing for LayoutLMv2 model (re-open) (#18313)

2022-07-27 06:38:40 -04:00

test_modeling_flax_common.py

[Flax] Add remat (gradient checkpointing) (#17843)

2022-07-01 18:33:54 +01:00

test_modeling_tf_common.py

Update serving code to enable saved_model=True (#18153)

2022-07-22 18:05:38 +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 some typos. (#17560)

2022-07-11 05:00:13 -04:00