huggingface/text-generation-inference
5
0
Fork 0
mirror of https://github.com/huggingface/text-generation-inference.git synced 2025-04-25 01:32:13 +00:00
Code Issues Packages Projects Releases Wiki Activity
1,066 Commits 229 Branches 59 Tags 24 MiB
add-test-for-warmup-and-kvcache
Commit Graph

3 Commits

Author SHA1 Message Date
David Holtz
541c476492 fix: cleanup ruff lints 2024-10-09 19:27:23 +00:00
David Holtz
1ddde382bd fix: only run test when cuda is available 2024-10-09 18:43:53 +00:00
David Holtz
a8108bc0da feat: add basic test for the warmup step and memory allocation of the kv cache 2024-10-09 13:49:11 +00:00