| .. |
|
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
|
Config: warning when saving generation kwargs in the model config (#28514)
|
2024-01-16 18:31:01 +00:00 |
|
models
|
Don't save processor_config.json if a processor has no extra attribute (#28584)
|
2024-01-19 09:59:14 +00:00 |
|
optimization
|
|
|
|
peft_integration
|
|
|
|
pipelines
|
Tokenizer kwargs in textgeneration pipe (#28362)
|
2024-01-15 16:52:18 +01:00 |
|
quantization
|
[GPTQ] Fix test (#28018)
|
2024-01-15 11:22:54 -05:00 |
|
repo_utils
|
|
|
|
sagemaker
|
|
|
|
tokenization
|
|
|
|
tools
|
|
|
|
trainer
|
Support DeepSpeed when using auto find batch size (#28088)
|
2024-01-10 06:03:13 -05:00 |
|
utils
|
improve dev setup comments and hints (#28495)
|
2024-01-15 18:36:40 +00: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
|
Generate: SinkCache can handle iterative prompts (#27907)
|
2023-12-08 20:02:20 +00:00 |
|
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
|
|
|
|
test_modeling_flax_utils.py
|
|
|
|
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
|
|
|
