transformers/tests
Yih-Dar 59cd9de39d
Byebye torch 1.10 (#28207)
* fix

* fix

---------

Co-authored-by: ydshieh <ydshieh@users.noreply.github.com>
2024-01-11 16:18:27 +01:00
..
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 [BUG] BarkEosPrioritizerLogitsProcessor eos_token_id use list, tensor size mismatch (#28201) 2024-01-10 11:46:49 +01:00
models Byebye torch 1.10 (#28207) 2024-01-11 16:18:27 +01:00
optimization
peft_integration
pipelines Byebye torch 1.10 (#28207) 2024-01-11 16:18:27 +01:00
quantization [Awq] Enable the possibility to skip quantization for some target modules (#27950) 2023-12-25 11:06:56 +01:00
repo_utils Allow # Ignore copy (#27328) 2023-12-07 10:00:08 +01:00
sagemaker
tokenization
tools
trainer Support DeepSpeed when using auto find batch size (#28088) 2024-01-10 06:03:13 -05:00
utils Fix the check of models supporting FA/SDPA not run (#28202) 2023-12-22 12:56:11 +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 Generate: SinkCache can handle iterative prompts (#27907) 2023-12-08 20:02:20 +00:00
test_configuration_common.py
test_configuration_utils.py F.scaled_dot_product_attention support (#26572) 2023-12-09 05:38:14 +09: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 Use mmap option to load_state_dict (#28331) 2024-01-10 09:57:30 +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 4D attention_mask support (#27539) 2023-12-17 11:08:04 +01:00
test_pipeline_mixin.py
test_sequence_feature_extraction_common.py
test_tokenization_common.py
test_tokenization_utils.py