huggingface/text-generation-inference
5
0
Fork 0
mirror of https://github.com/huggingface/text-generation-inference.git synced 2025-04-23 16:02:10 +00:00
Code Issues Packages Projects Releases Wiki Activity
510d1c76c8
text-generation-inference/integration-tests/models
History
Nicolas Patry 510d1c76c8 Prefix test - Different kind of load test to trigger prefix test bugs. (#2490) 
* Adding prefix test.

* [WIP] tmp dump of integration load tests.

* Remove other tensor creation.

* Fixed the radix tree.

Used a slice everywhere in radix.rs to keep the cheap Arc cloning
instead of recomputing the input_ids.

* Fix parsing

* Is it really flashinfer version ?

* Remove some comments.

* Revert the max prefix hit.

* Adding numpy to diff.

* Upgraded flashinfer.

* Upgrading some stuff.

* Are we done yet ?

* Minor fixup

* Remove 1 log and put back the other.

* Add comment for why slot 0 is OK.

* Mounting on the job.

* Get me a debug branch

* Debugging CIs is fun.

* Attempt #28

* wip

* Tmate.

* Praying.

* Updating VLM causal model with updated context.

* Important line got squashed.

* Tmate again.

* Fingers crossed.

* We want only 1 run of integration tests.....

---------

Co-authored-by: Guillaume LEGENDRE <glegendre01@gmail.com>
2024-09-25 06:14:07 +00:00
..
__snapshots__ Prefix test - Different kind of load test to trigger prefix test bugs. (#2490) 2024-09-25 06:14:07 +00:00
test_bloom_560m_sharded.py Add pytest release marker (#2114) 2024-09-24 03:52:50 +00:00
test_bloom_560m.py Add pytest release marker (#2114) 2024-09-24 03:52:50 +00:00
test_chat_llama.py Lots of improvements (Still 2 allocators) (#2449) 2024-09-25 06:13:11 +00:00
test_completion_prompts.py Prefix test - Different kind of load test to trigger prefix test bugs. (#2490) 2024-09-25 06:14:07 +00:00
test_flash_awq_sharded.py Add pytest release marker (#2114) 2024-09-24 03:52:50 +00:00
test_flash_awq.py Add pytest release marker (#2114) 2024-09-24 03:52:50 +00:00
test_flash_deepseek_v2.py Add support for Deepseek V2 (#2224) 2024-09-25 05:27:40 +00:00
test_flash_falcon.py Add pytest release marker (#2114) 2024-09-24 03:52:50 +00:00
test_flash_gemma2.py Softcapping for gemma2. (#2273) 2024-09-25 05:31:08 +00:00
test_flash_gemma_gptq.py Add pytest release marker (#2114) 2024-09-24 03:52:50 +00:00
test_flash_gemma.py Add pytest release marker (#2114) 2024-09-24 03:52:50 +00:00
test_flash_gpt2.py Add pytest release marker (#2114) 2024-09-24 03:52:50 +00:00
test_flash_grammar_llama.py fix: correctly index into mask when applying grammar (#1618) 2024-04-25 10:16:16 +03:00
test_flash_llama_exl2.py Fixing exl2 and other quanize tests again. (#2419) 2024-09-25 06:08:38 +00:00
test_flash_llama_fp8.py Further fixes. (#2426) 2024-09-25 06:09:22 +00:00
test_flash_llama_gptq.py GPTQ CI improvements (#2151) 2024-09-25 05:21:03 +00:00
test_flash_llama_marlin_24.py Improve the handling of quantized weights (#2250) 2024-09-25 05:27:40 +00:00
test_flash_llama_marlin.py Add pytest release marker (#2114) 2024-09-24 03:52:50 +00:00
test_flash_llama_prefix.py Prefix test - Different kind of load test to trigger prefix test bugs. (#2490) 2024-09-25 06:14:07 +00:00
test_flash_llama.py feat(server): only compute prefill logprobs when asked (#406) 2023-06-02 17:12:30 +02:00
test_flash_medusa.py Revamp medusa implementation so that every model can benefit. (#1588) 2024-04-25 09:13:03 +03:00
test_flash_mistral.py fix(router): fix openapi and add jsonschema validation (#1578) 2024-04-24 18:07:44 +03:00
test_flash_neox_sharded.py Add pytest release marker (#2114) 2024-09-24 03:52:50 +00:00
test_flash_neox.py Add pytest release marker (#2114) 2024-09-24 03:52:50 +00:00
test_flash_pali_gemma.py feat: add ruff and resolve issue (#2262) 2024-09-25 05:46:24 +00:00
test_flash_phi.py Add pytest release marker (#2114) 2024-09-24 03:52:50 +00:00
test_flash_qwen2.py Add pytest release marker (#2114) 2024-09-24 03:52:50 +00:00
test_flash_santacoder.py Add pytest release marker (#2114) 2024-09-24 03:52:50 +00:00
test_flash_starcoder2.py Add pytest release marker (#2114) 2024-09-24 03:52:50 +00:00
test_flash_starcoder_gptq.py Upgrading the tests to match the current workings. (#2423) 2024-09-25 06:08:38 +00:00
test_flash_starcoder.py Add pytest release marker (#2114) 2024-09-24 03:52:50 +00:00
test_grammar_llama.py Add pytest release marker (#2114) 2024-09-24 03:52:50 +00:00
test_grammar_response_format_llama.py Upgrade fbgemm (#2398) 2024-09-25 06:04:51 +00:00
test_idefics2.py Refactor dead code - Removing all flash_xxx.py files. (#2166) 2024-09-25 05:20:28 +00:00
test_idefics.py feat: add ruff and resolve issue (#2262) 2024-09-25 05:46:24 +00:00
test_llava_next.py Add pytest release marker (#2114) 2024-09-24 03:52:50 +00:00
test_lora_mistral.py feat: simple mistral lora integration tests (#2180) 2024-09-25 05:27:40 +00:00
test_mamba.py All integration tests back everywhere (too many failed CI). (#2428) 2024-09-25 06:10:59 +00:00
test_mpt.py Add pytest release marker (#2114) 2024-09-24 03:52:50 +00:00
test_mt0_base.py Add pytest release marker (#2114) 2024-09-24 03:52:50 +00:00
test_neox_sharded.py Add pytest release marker (#2114) 2024-09-24 03:52:50 +00:00
test_neox.py Add pytest release marker (#2114) 2024-09-24 03:52:50 +00:00
test_opt.py Fix the prefix for OPT model in opt_modelling.py #2370 (CI RUN) (#2371) 2024-09-25 05:55:39 +00:00
test_t5_sharded.py Add pytest release marker (#2114) 2024-09-24 03:52:50 +00:00
test_tools_llama.py Pr 2451 ci branch (#2454) 2024-09-25 06:10:59 +00:00