9935720c87
* Add support for repacking AWQ weights for GPTQ-Marlin So far we couldn't support AWQ because virtually all AWQ models use symmetric quantization, which GPTQ-Marlin did not suppors. GPTQ-Marlin has recently added support AWQ repacking and AWQ asymmetric quantization (zero_point=True). This change updates all GPTQ-Marlin kernels from upstream and wires up AWQ support. For now enabling AWQ using Marlin requires running TGI with `--quantize gptq`. * Enable Marlin for supported AWQ configurations by default This makes the AWQ -> GPTQ repack test redundant, since we are now testing this with the regular AWQ test. |
||
|---|---|---|
| .. | ||
| custom_kernels | ||
| exllama_kernels | ||
| exllamav2_kernels | ||
| marlin | ||
| tests | ||
| text_generation_server | ||
| .gitignore | ||
| Makefile | ||
| Makefile-awq | ||
| Makefile-eetq | ||
| Makefile-fbgemm | ||
| Makefile-flash-att | ||
| Makefile-flash-att-v2 | ||
| Makefile-lorax-punica | ||
| Makefile-selective-scan | ||
| Makefile-vllm | ||
| README.md | ||
| fbgemm_remove_unused.patch | ||
| fix_torch90a.sh | ||
| poetry.lock | ||
| pyproject.toml | ||
| requirements_cuda.txt | ||
| requirements_intel.txt | ||
| requirements_rocm.txt | ||
README.md
Text Generation Inference Python gRPC Server
A Python gRPC server for Text Generation Inference
Install
make install
Run
make run-dev