mirror of
https://github.com/huggingface/text-generation-inference.git
synced 2025-04-21 23:12:07 +00:00
a4e3e8c608
* 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>
16 lines
382 B
TOML
16 lines
382 B
TOML
[tool.poetry]
|
|
name = "text-generation-integration-tests"
|
|
version = "2.0.1"
|
|
description = "Text Generation Inference integration tests"
|
|
authors = ["Nicolas Patry <nicolas@huggingface.co>"]
|
|
|
|
[tool.poetry.dependencies]
|
|
pydantic = "> 2, < 3"
|
|
python = ">=3.10,<3.13"
|
|
syrupy = "^4.7.1"
|
|
text-generation = "^0.6.0"
|
|
pytest = "^7.4.0"
|
|
pytest-asyncio = "^0.21.1"
|
|
docker = "^7"
|
|
numpy = "^1.20"
|