text-generation-inference/server/text_generation_server
Mohit Sharma ff905aeff3 Update ROCM libs and improvements (#2579)
* style

* update torch

* ix issues

* fix clone

* revert mkl

* added custom PA

* style

* fix style

* style

* hide env vart

* fix mixtral model

* add skinny kernel and merge fixes

* fixed style

* fix issue for sliding window models

* addressed review comments

* fix import

* improved error messag

* updated default value

* remove import

* fix imports after rebase

* float16 dep

* improve dockerfile

* cleaned dockerfile
2024-10-25 09:01:04 +00:00
..
adapters feat: add ruff and resolve issue (#2262) 2024-09-25 05:46:24 +00:00
layers Update ROCM libs and improvements (#2579) 2024-10-25 09:01:04 +00:00
models Update ROCM libs and improvements (#2579) 2024-10-25 09:01:04 +00:00
pb chore: add pre-commit (#1569) 2024-04-24 15:32:02 +03:00
utils Micro cleanup. (#2555) 2024-10-25 08:53:47 +00: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 Pass the max_batch_total_tokens to causal_lm 2024-10-23 08:28:26 +00:00
habana_quantization_env.py Remove all references to habana_quantization_toolkit for 1.18 (#229) 2024-10-18 10:59:59 +02:00
interceptor.py Make Gaudi adapt to the tgi 2.3.0 2024-09-26 06:04:55 +00:00
server.py Make Gaudi adapt to the tgi 2.3.0 2024-09-26 06:04:55 +00:00
tgi_service.py Make Gaudi adapt to the tgi 2.3.0 2024-09-26 06:04:55 +00:00
tracing.py Add OTLP Service Name Environment Variable (#2076) 2024-09-24 03:51:26 +00:00