hf_text-generation-inference/launcher
Daniël de Kok 5b6b74e21d
Improve support for GPUs with capability < 8 (#2575)
* Improve support for GPUs with capability < 8

- For models that cannot use flashinfer, use flash-attn v1 + paged
  attention for models with a compute capability older than 8.
- Disable prefix caching when using paged attention.
- When using flash-attn v1, pass the key/value, rather than the
  cache, since v1 cannot use block tables.

* nix: add flash-attn-v1 to the server environment

* Move disabling prefix caching into the block of exceptions

* Capability as `usize`s
2024-09-27 16:19:42 +02:00
..
src Improve support for GPUs with capability < 8 (#2575) 2024-09-27 16:19:42 +02:00
Cargo.toml Improve support for GPUs with capability < 8 (#2575) 2024-09-27 16:19:42 +02:00
build.rs chore(github): add templates (#264) 2023-05-02 15:43:19 +02:00