..
benchmark
[Test refactor 1/5] Per-folder tests reorganization ( #15725 )
2022-02-23 15:46:28 -05:00
bettertransformer
Add methods to PreTrainedModel to use PyTorch's BetterTransformer ( #21259 )
2023-04-27 11:03:42 +02:00
bnb
[tests] fix bitsandbytes import issue ( #24151 )
2023-06-09 21:53:11 -07:00
deepspeed
accelerate deepspeed and gradient accumulation integrate ( #23236 )
2023-05-31 15:16:22 +05:30
extended
[tests] switch to torchrun ( #22712 )
2023-04-12 08:25:45 -07:00
fixtures
[WIP] add SpeechT5 model ( #18922 )
2023-02-03 12:43:46 -05:00
generation
Replace python random with torch.rand to enable dynamo.export ( #24434 )
2023-06-23 08:17:21 -04:00
models
Revert "Fix gradient checkpointing + fp16 autocast for most models" ( #24420 )
2023-06-22 16:11:27 +02:00
onnx
Fix issue introduced in PR #23163 ( #23363 )
2023-05-15 11:38:44 +02:00
optimization
Make schedulers picklable by making lr_lambda fns global ( #21768 )
2023-03-02 12:08:43 -05:00
pipelines
Skip test_conditional_generation_pt_pix2struct
in Past CI (torch < 1.11) ( #24417 )
2023-06-22 15:34:13 +02:00
repo_utils
Fix expected value in tests of the test fetcher ( #24077 )
2023-06-07 11:38:56 -04:00
sagemaker
Avoid invalid escape sequences, use raw strings ( #22936 )
2023-04-25 09:17:56 -04:00
tokenization
Update quality tooling for formatting ( #21480 )
2023-02-06 18:10:56 -05:00
tools
Fix image segmentation tool bug ( #23897 )
2023-06-15 08:09:31 -04:00
trainer
Refactor hyperparameter search backends ( #24384 )
2023-06-22 14:28:25 -04:00
utils
Migrate doc files to Markdown. ( #24376 )
2023-06-20 18:07:47 -04:00
__init__.py
GPU text generation: mMoved the encoded_prompt to correct device
2020-01-06 15:11:12 +01:00
test_backbone_common.py
Add TimmBackbone model ( #22619 )
2023-06-06 17:11:30 +01:00
test_configuration_common.py
Split common test from core tests ( #24284 )
2023-06-15 07:30:24 -04:00
test_configuration_utils.py
Split common test from core tests ( #24284 )
2023-06-15 07:30:24 -04:00
test_feature_extraction_common.py
Split common test from core tests ( #24284 )
2023-06-15 07:30:24 -04:00
test_feature_extraction_utils.py
Split common test from core tests ( #24284 )
2023-06-15 07:30:24 -04:00
test_image_processing_common.py
Split common test from core tests ( #24284 )
2023-06-15 07:30:24 -04:00
test_image_processing_utils.py
Split common test from core tests ( #24284 )
2023-06-15 07:30:24 -04:00
test_image_transforms.py
Bug fix - flip_channel_order for channels first images ( #23701 )
2023-05-31 17:12:27 +01:00
test_modeling_common.py
Revert "Fix gradient checkpointing + fp16 autocast for most models" ( #24420 )
2023-06-22 16:11:27 +02:00
test_modeling_flax_common.py
Split common test from core tests ( #24284 )
2023-06-15 07:30:24 -04:00
test_modeling_flax_utils.py
Split common test from core tests ( #24284 )
2023-06-15 07:30:24 -04:00
test_modeling_tf_common.py
Add test for proper TF input signatures ( #24320 )
2023-06-16 17:03:13 +01:00
test_modeling_tf_utils.py
Split common test from core tests ( #24284 )
2023-06-15 07:30:24 -04:00
test_modeling_utils.py
Tied weights load ( #24310 )
2023-06-16 10:55:42 -04:00
test_pipeline_mixin.py
Update tiny models for pipeline testing. ( #24364 )
2023-06-20 14:43:10 +02:00
test_sequence_feature_extraction_common.py
Apply ruff flake8-comprehensions ( #21694 )
2023-02-22 09:14:54 +01:00
test_tokenization_common.py
Split common test from core tests ( #24284 )
2023-06-15 07:30:24 -04:00
test_tokenization_utils.py
Split common test from core tests ( #24284 )
2023-06-15 07:30:24 -04:00