HuggingFace_transformer

Files

Yao Matrix 5534b80b7f enable xpu in test_trainer (#37774 )

* enable xpu in test_trainer

Signed-off-by: YAO Matrix <matrix.yao@intel.com>

* fix style

Signed-off-by: YAO Matrix <matrix.yao@intel.com>

* enhance _device_agnostic_dispatch to cover value

Signed-off-by: Yao Matrix <matrix.yao@intel.com>

* add default values for torch not available case

Signed-off-by: YAO Matrix <matrix.yao@intel.com>

---------

Signed-off-by: YAO Matrix <matrix.yao@intel.com>
Signed-off-by: Yao Matrix <matrix.yao@intel.com>

2025-05-06 17:13:35 +02:00

__init__.py

[Test refactor 1/5] Per-folder tests reorganization (#15725 )

2022-02-23 15:46:28 -05:00

test_data_collator.py

add FlashAttentionKwargs and seq_idx to flat collator (#36456 )

2025-04-16 15:45:03 +02:00

test_trainer_callback.py

fix: prevent second save in the end of training if last step was saved already (#36219 )

2025-02-20 17:38:52 +01:00

test_trainer_distributed_loss.py

Fix multi gpu loss sync condition, add doc and test (#35743 )