..
benchmark
[Test refactor 1/5] Per-folder tests reorganization ( #15725 )
2022-02-23 15:46:28 -05:00
deepspeed
[trainer/deepspeed] load_best_model (reimplement re-init) ( #17151 )
2022-06-02 09:14:21 -07: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
Add Gated-SiLU to T5 ( #17420 )
2022-06-03 10:56:37 +02:00
onnx
Add OnnxConfig for SqueezeBert iss17314 ( #17315 )
2022-06-01 06:16:15 -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
[Test refactor 1/5] Per-folder tests reorganization ( #15725 )
2022-02-23 15:46:28 -05:00
trainer
Support compilation via Torchdynamo, AOT Autograd, NVFuser ( #17308 )
2022-05-25 11:16:09 -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
Fix MP and CPU offload tests for Funnel and GPT-Neo ( #17503 )
2022-06-01 09:59:40 -04: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
Rewrite TensorFlow train_step and test_step ( #17057 )
2022-05-17 14:36:23 +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
[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