| .. |
|
benchmark
|
|
|
|
bettertransformer
|
|
|
|
deepspeed
|
Fix initialization for missing parameters in from_pretrained under ZeRO-3 (#28245)
|
2024-01-09 14:58:21 +00:00 |
|
extended
|
|
|
|
fixtures
|
|
|
|
fsdp
|
fix resuming from ckpt when using FSDP with FULL_STATE_DICT (#27891)
|
2023-12-16 19:41:43 +05:30 |
|
generation
|
Mark test_constrained_beam_search_generate as flaky (#28757)
|
2024-01-29 15:22:25 +00:00 |
|
models
|
PatchtTST and PatchTSMixer fixes (#28083)
|
2024-01-29 10:09:26 +00:00 |
|
optimization
|
|
|
|
peft_integration
|
|
|
|
pipelines
|
[Whisper] Finalize batched SOTA long-form generation (#27658)
|
2024-01-19 14:04:17 +02:00 |
|
quantization
|
[GPTQ] Fix test (#28018)
|
2024-01-15 11:22:54 -05:00 |
|
repo_utils
|
|
|
|
sagemaker
|
|
|
|
tokenization
|
|
|
|
tools
|
|
|
|
trainer
|
Avoid root logger's level being changed (#28638)
|
2024-01-22 14:45:30 +01:00 |
|
utils
|
Add serialization logic to pytree types (#27871)
|
2024-01-29 10:41:20 +01:00 |
|
__init__.py
|
|
|
|
test_backbone_common.py
|
Align backbone stage selection with out_indices & out_features (#27606)
|
2023-12-20 18:33:17 +00:00 |
|
test_cache_utils.py
|
|
|
|
test_configuration_common.py
|
|
|
|
test_configuration_utils.py
|
Config: warning when saving generation kwargs in the model config (#28514)
|
2024-01-16 18:31:01 +00:00 |
|
test_feature_extraction_common.py
|
|
|
|
test_feature_extraction_utils.py
|
|
|
|
test_image_processing_common.py
|
Fix a couple of typos and add an illustrative test (#26941)
|
2023-12-11 15:51:51 +00:00 |
|
test_image_processing_utils.py
|
|
|
|
test_image_transforms.py
|
|
|
|
test_modeling_common.py
|
Fix SDPA tests (#28552)
|
2024-01-17 17:29:18 +01:00 |
|
test_modeling_flax_common.py
|
[Flax] Update no init test for Flax v0.7.1 (#28735)
|
2024-01-26 18:20:39 +00:00 |
|
test_modeling_flax_utils.py
|
Enable safetensors conversion from PyTorch to other frameworks without the torch requirement (#27599)
|
2024-01-23 10:28:23 +01:00 |
|
test_modeling_tf_common.py
|
Replace build() with build_in_name_scope() for some TF tests (#28046)
|
2023-12-14 17:42:25 +00:00 |
|
test_modeling_tf_utils.py
|
Replace build() with build_in_name_scope() for some TF tests (#28046)
|
2023-12-14 17:42:25 +00:00 |
|
test_modeling_utils.py
|
Use LoggingLevel context manager in 3 tests (#28575)
|
2024-01-18 13:41:25 +00:00 |
|
test_pipeline_mixin.py
|
|
|
|
test_processing_common.py
|
Don't save processor_config.json if a processor has no extra attribute (#28584)
|
2024-01-19 09:59:14 +00:00 |
|
test_sequence_feature_extraction_common.py
|
|
|
|
test_tokenization_common.py
|
[ TokenizationUtils] Fix add_special_tokens when the token is already there (#28520)
|
2024-01-16 16:36:29 +01:00 |
|
test_tokenization_utils.py
|
|
|
