Patrick von Platen 8604dd308d [SDPA] Make sure attn mask creation is always done on CPU (#28400) ... * [SDPA] Make sure attn mask creation is always done on CPU * Update docker to 2.1.1 * revert test change		2024-01-09 11:05:19 +01:00
..
transformers-all-latest-gpu	[SDPA] Make sure attn mask creation is always done on CPU (#28400)	2024-01-09 11:05:19 +01:00
transformers-doc-builder	Don't install pytorch-quantization in Doc Builder docker file (#26622)	2023-10-05 16:57:50 +02:00
transformers-gpu	TF: TF 2.10 unpin + related onnx test skips (#18995)	2022-09-12 19:30:27 +01:00
transformers-past-gpu	Byebye pytorch 1.9 (#24080)	2023-06-16 16:38:23 +02:00
transformers-pytorch-amd-gpu	Add deepspeed test to amd scheduled CI (#27633)	2023-12-11 16:33:36 +01:00
transformers-pytorch-deepspeed-amd-gpu	Add deepspeed test to amd scheduled CI (#27633)	2023-12-11 16:33:36 +01:00
transformers-pytorch-deepspeed-latest-gpu	Add deepspeed test to amd scheduled CI (#27633)	2023-12-11 16:33:36 +01:00
transformers-pytorch-deepspeed-nightly-gpu	Update CUDA versions for DeepSpeed (#27853)	2023-12-05 16:15:21 -05:00
transformers-pytorch-gpu	[SDPA] Make sure attn mask creation is always done on CPU (#28400)	2024-01-09 11:05:19 +01:00
transformers-pytorch-tpu	Rename master to main for notebooks links and leftovers (#16397)	2022-03-25 09:12:23 -04:00
transformers-tensorflow-gpu	Update TF pin in docker image (#25343)	2023-08-07 12:32:34 +02:00