transformers
2c1eebc1 - Fix SDPA tests (#28552)

Commit
2 years ago
Fix SDPA tests (#28552) * skip bf16 test if not supported by device * fix * fix bis * use is_torch_bf16_available_on_device * use is_torch_fp16_available_on_device * fix & use public llama * use 1b model * fix flacky test --------- Co-authored-by: Your Name <you@example.com>
Author
Parents
Loading