text-generation-inference/server/text_generation_server/layers
Wang, Yi b64c70c9e7
Cpu tgi (#1936)
* add CPU tgi support

Signed-off-by: Wang, Yi A <yi.a.wang@intel.com>

* ipex distributed ops support

Signed-off-by: Wang, Yi A <yi.a.wang@intel.com>

---------

Signed-off-by: Wang, Yi A <yi.a.wang@intel.com>
Co-authored-by: Funtowicz Morgan <mfuntowicz@users.noreply.github.com>
2024-06-25 12:21:29 +02:00
..
attention Cpu tgi (#1936) 2024-06-25 12:21:29 +02:00
awq Refactor layers. (#1866) 2024-05-13 12:44:30 +02:00
gptq Fix text-generation-server quantize (#2103) 2024-06-21 15:28:51 +02:00
__init__.py MLPSpeculator. (#1865) 2024-05-14 12:33:18 +02:00
bnb.py [Bug Fix] Update torch import reference in bnb quantization (#1902) 2024-05-15 21:08:32 +02:00
conv.py Refactor layers. (#1866) 2024-05-13 12:44:30 +02:00
eetq.py Refactor layers. (#1866) 2024-05-13 12:44:30 +02:00
exl2.py Add support for exl2 quantization 2024-05-30 11:28:05 +02:00
fp8.py Refactor layers. (#1866) 2024-05-13 12:44:30 +02:00
layernorm.py Cpu tgi (#1936) 2024-06-25 12:21:29 +02:00
linear.py Add support for GPTQ Marlin (#2052) 2024-06-14 09:45:42 +02:00
marlin.py Add support for GPTQ Marlin (#2052) 2024-06-14 09:45:42 +02:00
medusa.py fix: use path inside of speculator config (#1935) 2024-05-22 20:46:29 +02:00
mlp.py MLPSpeculator. (#1865) 2024-05-14 12:33:18 +02:00
rotary.py Cpu tgi (#1936) 2024-06-25 12:21:29 +02:00
speculative.py MLPSpeculator. (#1865) 2024-05-14 12:33:18 +02:00
tensor_parallel.py Cpu tgi (#1936) 2024-06-25 12:21:29 +02:00