hf_text-generation-inference/integration-tests/models
Nicolas Patry 38fcafcf96
Adding a test for FD. (#2516)
* Adding a test for FD.

* Fixing flashdecoding (empty batch doesn't work).

* Fixing the invalid popping.

* Fixing radix with block_size > 1

* Last reference.

* Use an actual hash.

* Update hash for slice.len() == 1

* Update the locks.

* Increasing docker timeout.
2024-09-16 17:00:54 +02:00
..
__snapshots__ Adding a test for FD. (#2516) 2024-09-16 17:00:54 +02:00
test_bloom_560m.py
test_bloom_560m_sharded.py
test_chat_llama.py
test_completion_prompts.py
test_flash_awq.py
test_flash_awq_sharded.py
test_flash_deepseek_v2.py
test_flash_falcon.py
test_flash_gemma.py
test_flash_gemma2.py
test_flash_gemma_gptq.py
test_flash_gpt2.py
test_flash_grammar_llama.py
test_flash_llama.py
test_flash_llama_exl2.py
test_flash_llama_fp8.py
test_flash_llama_gptq.py
test_flash_llama_marlin.py
test_flash_llama_marlin_24.py
test_flash_llama_prefix.py
test_flash_llama_prefix_flashdecoding.py Adding a test for FD. (#2516) 2024-09-16 17:00:54 +02:00
test_flash_medusa.py
test_flash_mistral.py
test_flash_mixtral.py Add tests for Mixtral (#2520) 2024-09-16 12:39:18 +02:00
test_flash_neox.py
test_flash_neox_sharded.py
test_flash_pali_gemma.py
test_flash_phi.py
test_flash_qwen2.py
test_flash_santacoder.py
test_flash_starcoder.py
test_flash_starcoder2.py
test_flash_starcoder_gptq.py
test_grammar_llama.py
test_grammar_response_format_llama.py
test_idefics.py
test_idefics2.py
test_llava_next.py
test_lora_mistral.py
test_mamba.py
test_mpt.py
test_mt0_base.py
test_neox.py
test_neox_sharded.py
test_opt.py
test_t5_sharded.py
test_tools_llama.py