hf_text-generation-inference/server/text_generation_server
Daniël de Kok 9ffe1f1e67
Do not initialize scratch space when there are no ExLlamaV2 layers (#2015)
# What does this PR do?

Do not attempt to allocate ExLlamaV2 scratch buffers when there are no
ExLlama2 layers. Avoids a crash in warmup for models that cannot use
exllama when ExLlamaV2 is installed.

## Before submitting
- [ ] This PR fixes a typo or improves the docs (you can dismiss the
other checks if that's the case).
- [ ] Did you read the [contributor
guideline](https://github.com/huggingface/transformers/blob/main/CONTRIBUTING.md#start-contributing-pull-requests),
      Pull Request section?
- [ ] Was this discussed/approved via a Github issue or the
[forum](https://discuss.huggingface.co/)? Please add a link
      to it if that's the case.
- [ ] Did you make sure to update the documentation with your changes?
Here are the
[documentation
guidelines](https://github.com/huggingface/transformers/tree/main/docs),
and
[here are tips on formatting
docstrings](https://github.com/huggingface/transformers/tree/main/docs#writing-source-documentation).
- [ ] Did you write any new necessary tests?


## Who can review?

Anyone in the community is free to review the PR once the tests have
passed. Feel free to tag
members/contributors who may be interested in your PR.

<!-- Your PR will be replied to more quickly if you can figure out the
right person to tag with @


@OlivierDehaene OR @Narsil

 -->
2024-06-05 10:45:47 +02:00
..
layers Do not initialize scratch space when there are no ExLlamaV2 layers (#2015) 2024-06-05 10:45:47 +02:00
models Fix Phi-2 with `tp>1` (#2003) 2024-06-04 14:26:07 +02:00
pb chore: add pre-commit (#1569) 2024-02-16 11:58:58 +01:00
utils Support GPTQ models with column-packed up/gate tensor (#2006) 2024-06-04 19:37:49 +02:00
__init__.py feat(clients): Python client (#103) 2023-03-07 18:52:22 +01:00
cache.py fix(server): decrease memory fragmentation (#557) 2023-07-06 14:28:33 +02:00
cli.py Add support for exl2 quantization 2024-05-30 11:28:05 +02:00
interceptor.py v2.0.0 (#1736) 2024-04-12 18:38:34 +02:00
server.py Add support for exl2 quantization 2024-05-30 11:28:05 +02:00
tracing.py feat(clients): Python client (#103) 2023-03-07 18:52:22 +01:00