text-generation-inference/server
Dhruv Srikanth b1d370e062 Update torch import reference in bnb quantization (#1902)
# What does this PR do?

Fixes `Import Error` occurring from mismatch of usage between
torch.nn.Module and nn.Module.
2024-07-17 05:36:58 +00:00
..
custom_kernels chore: add pre-commit (#1569) 2024-04-24 15:32:02 +03:00
exllama_kernels chore: add pre-commit (#1569) 2024-04-24 15:32:02 +03:00
exllamav2_kernels chore: add pre-commit (#1569) 2024-04-24 15:32:02 +03:00
tests Refactor layers. (#1866) 2024-07-17 05:36:58 +00:00
text_generation_server Update torch import reference in bnb quantization (#1902) 2024-07-17 05:36:58 +00:00
.gitignore Impl simple mamba model (#1480) 2024-04-23 11:45:11 +03:00
dill-0.3.7-patch.sh Hgraph dill patch (#131) 2024-04-26 11:08:15 +02:00
dill-0.3.8-patch.sh A patch to address HPU Graphs issue with DILL 2024-05-06 09:15:46 +03:00
Makefile fix: fix CohereForAI/c4ai-command-r-plus (#1707) 2024-04-25 17:51:35 +03:00
Makefile-awq chore: add pre-commit (#1569) 2024-04-24 15:32:02 +03:00
Makefile-eetq Upgrade EETQ (Fixes the cuda graphs). (#1729) 2024-04-25 17:58:27 +03:00
Makefile-flash-att chore: add pre-commit (#1569) 2024-04-24 15:32:02 +03:00
Makefile-flash-att-v2 fix: fix CohereForAI/c4ai-command-r-plus (#1707) 2024-04-25 17:51:35 +03:00
Makefile-selective-scan chore: add pre-commit (#1569) 2024-04-24 15:32:02 +03:00
Makefile-vllm (chore): torch 2.3.0 (#1833) 2024-06-10 14:12:46 +03:00
poetry.lock Granite support? (#1882) 2024-07-17 05:36:58 +00:00
pyproject.toml Update to SynapseAI 1.16.0 (#167) 2024-07-03 11:08:56 +02:00
README.md chore: add pre-commit (#1569) 2024-04-24 15:32:02 +03:00
requirements_cuda.txt Granite support? (#1882) 2024-07-17 05:36:58 +00:00
requirements_rocm.txt Granite support? (#1882) 2024-07-17 05:36:58 +00:00
requirements.txt Update to SynapseAI 1.16.0 (#167) 2024-07-03 11:08:56 +02:00

Text Generation Inference Python gRPC Server

A Python gRPC server for Text Generation Inference

Install

make install

Run

make run-dev