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