hf_text-generation-inference/server
Nick Hill 686cc66717
fix(server): Check for device type correctly when determining initial padding (#16)
AFAIK there is no torch device type called "gpu".
2022-12-30 19:30:42 +01:00
..
tests fix(server): Fix stop sequences (#11) 2022-12-16 16:03:39 +01:00
text_generation fix(server): Check for device type correctly when determining initial padding (#16) 2022-12-30 19:30:42 +01:00
.gitignore feat(server): Support all AutoModelForCausalLM on a best effort basis 2022-10-28 19:24:00 +02:00
Makefile feat(server): Add model tests (#6) 2022-12-08 18:49:33 +01:00
README.md feat(server): Use safetensors 2022-10-22 20:00:15 +02:00
poetry.lock feat(server): Add model tests (#6) 2022-12-08 18:49:33 +01:00
pyproject.toml feat(server): Add model tests (#6) 2022-12-08 18:49:33 +01:00

README.md

BLOOM Inference Python gRPC Server

A Python gRPC server for BLOOM Inference

Install

make install

Run

make run-dev