Cyril Vallez 0cfbf9c95b Force torch>=2.6 with torch.load to avoid vulnerability issue (#37785) ...

* fix all main files

* fix test files

* oups forgot modular

* add link

* update message

2025-04-25 16:57:09 +02:00

..

__init__.py

…

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)

2025-02-12 15:41:31 +01:00

test_trainer_distributed.py

Use Python 3.9 syntax in tests (#37343)

2025-04-08 14:12:08 +02:00

test_trainer_fsdp.py

Remove old code for PyTorch, Accelerator and tokenizers (#37234)

2025-04-10 20:54:21 +02:00

test_trainer_seq2seq.py

Use Python 3.9 syntax in tests (#37343)

2025-04-08 14:12:08 +02:00

test_trainer_tpu.py

Use Python 3.9 syntax in tests (#37343)

2025-04-08 14:12:08 +02:00

test_trainer_utils.py

Use Python 3.9 syntax in tests (#37343)

2025-04-08 14:12:08 +02:00

test_trainer.py

Force torch>=2.6 with torch.load to avoid vulnerability issue (#37785)

2025-04-25 16:57:09 +02:00