1da07e85aa
# What does this PR do? This adds a non flash version of MPT. Flash is harder because we need to create a bias ready cuda kernel of flash attention. Fixes https://github.com/huggingface/text-generation-inference/issues/361 Fixes https://github.com/huggingface/text-generation-inference/issues/491 Fixes https://github.com/huggingface/text-generation-inference/issues/290 |
||
|---|---|---|
| .. | ||
| test_bloom_560m | ||
| test_bloom_560m_sharded | ||
| test_flash_falcon | ||
| test_flash_llama | ||
| test_flash_neox | ||
| test_flash_neox_sharded | ||
| test_flash_santacoder | ||
| test_flash_starcoder | ||
| test_mpt | ||
| test_mt0_base | ||
| test_neox | ||
| test_neox_sharded | ||
| test_t5_sharded | ||