Daniël de Kok 289aa48554 Move JSON grammar -> regex grammar conversion to the router (#2772) ... * Move JSON grammar -> regex grammar conversion to the router This change moves the JSON grammar -> regex grammar conversion to the router by adding a dependency on the `outlines-core` Rust crate. In contrast to the Python implementation, the conversions are not LRU-cached since they seem to be fast enough: simple schema time: [5.8293 µs 5.8307 µs 5.8320 µs] change: [-13.166% -12.884% -12.641%] (p = 0.00 < 0.05) Performance has improved. complex schema time: [14.875 µs 14.881 µs 14.887 µs] change: [-2.1637% -1.9914% -1.7852%] (p = 0.00 < 0.05) Performance has improved. Using the schemas from: https://github.com/dottxt-ai/outlines-core/blob/main/benchmarks/bench_json_schema.py		2024-11-25 18:47:34 +01:00
..
__snapshots__	Move JSON grammar -> regex grammar conversion to the router (#2772)	2024-11-25 18:47:34 +01:00
test_bloom_560m_sharded.py	Add pytest release marker (#2114)	2024-06-25 16:53:20 +02:00
test_bloom_560m.py	Avoiding timeout for bloom tests. (#2693)	2024-10-26 05:35:28 +02:00
test_chat_llama.py	Lots of improvements (Still 2 allocators) (#2449)	2024-08-29 16:29:01 +02:00
test_completion_prompts.py	Stream options. (#2533)	2024-09-19 20:50:37 +02:00
test_compressed_tensors_w8a8_int_dynamic_weight.py	Add support for compressed-tensors w8a8 int checkpoints (#2745)	2024-11-18 17:20:31 +01:00
test_compressed_tensors_w8a8_int.py	Add support for compressed-tensors w8a8 int checkpoints (#2745)	2024-11-18 17:20:31 +01:00
test_compressed_tensors_w8an_fp.py	Add initial support for compressed-tensors checkpoints (#2732)	2024-11-10 13:54:07 +01:00
test_compressed_tensors_wna16_int_24.py	Add support for wNa16 int 2:4 compressed-tensors checkpoints (#2758)	2024-11-20 18:25:23 +01:00
test_compressed_tensors_wna16_int.py	Add initial support for compressed-tensors checkpoints (#2732)	2024-11-10 13:54:07 +01:00
test_flash_awq_sharded.py	Add pytest release marker (#2114)	2024-06-25 16:53:20 +02:00
test_flash_awq.py	Add pytest release marker (#2114)	2024-06-25 16:53:20 +02:00
test_flash_deepseek_v2.py	Add support for Deepseek V2 (#2224)	2024-07-19 17:23:20 +02:00
test_flash_falcon.py	Add pytest release marker (#2114)	2024-06-25 16:53:20 +02:00
test_flash_gemma2.py	Softcapping for gemma2. (#2273)	2024-07-22 18:27:10 +02:00
test_flash_gemma_gptq.py	Add pytest release marker (#2114)	2024-06-25 16:53:20 +02:00
test_flash_gemma.py	Intel ci (#2630)	2024-10-10 16:51:57 +02:00
test_flash_gpt2.py	Add pytest release marker (#2114)	2024-06-25 16:53:20 +02:00
test_flash_grammar_llama.py	fix: correctly index into mask when applying grammar (#1618)	2024-03-01 18:22:01 +01:00
test_flash_llama_exl2.py	Fixing exl2 and other quanize tests again. (#2419)	2024-08-15 11:12:51 +02:00
test_flash_llama_fp8_kv_cache.py	Add support for FP8 KV cache scales (#2628)	2024-10-24 16:36:18 +02:00
test_flash_llama_fp8.py	Further fixes. (#2426)	2024-08-16 13:21:44 +02:00
test_flash_llama_gptq.py	GPTQ CI improvements (#2151)	2024-07-05 14:12:16 +02:00
test_flash_llama_marlin_24.py	Improve the handling of quantized weights (#2250)	2024-07-19 09:37:39 +02:00
test_flash_llama_marlin.py	Add pytest release marker (#2114)	2024-06-25 16:53:20 +02:00
test_flash_llama_prefix_flashdecoding.py	Adding a test for FD. (#2516)	2024-09-16 17:00:54 +02:00
test_flash_llama_prefix.py	Fix truffle (#2514)	2024-09-11 22:45:19 +02:00
test_flash_llama.py	Intel ci (#2630)	2024-10-10 16:51:57 +02:00
test_flash_medusa.py	Revamp medusa implementation so that every model can benefit. (#1588)	2024-02-26 19:49:28 +01:00
test_flash_mistral.py	fix(router): fix openapi and add jsonschema validation (#1578)	2024-02-21 11:05:32 +01:00
test_flash_mixtral_awq.py	Add support for fused MoE Marlin for AWQ (#2616)	2024-10-08 11:56:41 +02:00
test_flash_mixtral_gptq.py	Test Marlin MoE with desc_act=true (#2622)	2024-10-21 12:50:35 +02:00
test_flash_mixtral.py	Add tests for Mixtral (#2520)	2024-09-16 12:39:18 +02:00
test_flash_neox_sharded.py	Add pytest release marker (#2114)	2024-06-25 16:53:20 +02:00
test_flash_neox.py	Add pytest release marker (#2114)	2024-06-25 16:53:20 +02:00
test_flash_pali_gemma.py	feat: prefill chunking (#2600)	2024-10-16 12:49:33 +02:00
test_flash_phi35_moe.py	Fix Phi 3.5 MoE tests (#2684)	2024-10-24 15:21:50 +02:00
test_flash_phi.py	Add pytest release marker (#2114)	2024-06-25 16:53:20 +02:00
test_flash_qwen2_vl.py	fix cuda graphs for qwen2-vl (#2708)	2024-11-01 03:05:34 +01:00
test_flash_qwen2.py	Add pytest release marker (#2114)	2024-06-25 16:53:20 +02:00
test_flash_santacoder.py	Add pytest release marker (#2114)	2024-06-25 16:53:20 +02:00
test_flash_starcoder2.py	Add pytest release marker (#2114)	2024-06-25 16:53:20 +02:00
test_flash_starcoder_gptq.py	Green main (#2697)	2024-10-28 04:59:32 +01:00
test_flash_starcoder.py	Add pytest release marker (#2114)	2024-06-25 16:53:20 +02:00
test_grammar_llama.py	Add pytest release marker (#2114)	2024-06-25 16:53:20 +02:00
test_grammar_response_format_llama.py	Move JSON grammar -> regex grammar conversion to the router (#2772)	2024-11-25 18:47:34 +01:00
test_idefics2.py	feat: prefill chunking (#2600)	2024-10-16 12:49:33 +02:00
test_idefics.py	feat: prefill chunking (#2600)	2024-10-16 12:49:33 +02:00
test_llava_next.py	feat: prefill chunking (#2600)	2024-10-16 12:49:33 +02:00
test_lora_mistral.py	feat: simple mistral lora integration tests (#2180)	2024-07-15 09:16:15 -04:00
test_mamba.py	We can have a tokenizer anywhere. (#2527)	2024-10-28 05:00:24 +01:00
test_mllama.py	Green main (#2697)	2024-10-28 04:59:32 +01:00
test_mpt.py	Add pytest release marker (#2114)	2024-06-25 16:53:20 +02:00
test_mt0_base.py	Add pytest release marker (#2114)	2024-06-25 16:53:20 +02:00
test_neox_sharded.py	Add pytest release marker (#2114)	2024-06-25 16:53:20 +02:00
test_neox.py	Add pytest release marker (#2114)	2024-06-25 16:53:20 +02:00
test_opt.py	Fix the prefix for OPT model in opt_modelling.py #2370 (CI RUN) (#2371)	2024-08-07 23:14:02 -04:00
test_t5_sharded.py	Add pytest release marker (#2114)	2024-06-25 16:53:20 +02:00
test_tools_llama.py	Move JSON grammar -> regex grammar conversion to the router (#2772)	2024-11-25 18:47:34 +01:00