fxmarty 2c1eebc121 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>

2024-01-17 17:29:18 +01:00

..

__init__.py

LLaMA Implementation (#21955)

2023-03-16 09:00:53 -04:00

test_modeling_flax_llama.py

Add Llama Flax Implementation (#24587)

2023-12-07 07:05:00 +01:00

test_modeling_llama.py

Fix SDPA tests (#28552)

2024-01-17 17:29:18 +01:00

test_tokenization_llama.py

[Styling] stylify using ruff (#27144)

2023-11-16 17:43:19 +01:00