hf_text-generation-inference/server/text_generation
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
..
models fix(server): Check for device type correctly when determining initial padding (#16) 2022-12-30 19:30:42 +01:00
pb feat(server): Support all AutoModelForCausalLM on a best effort basis 2022-10-28 19:24:00 +02:00
__init__.py feat(server): Support all AutoModelForCausalLM on a best effort basis 2022-10-28 19:24:00 +02:00
cache.py feat(server): Support AutoModelForSeq2SeqLM 2022-11-04 18:03:04 +01:00
cli.py feat(server): Support all AutoModelForCausalLM on a best effort basis 2022-10-28 19:24:00 +02:00
server.py feat(server): Support AutoModelForSeq2SeqLM 2022-11-04 18:03:04 +01:00
utils.py fix(server): Fix stop sequences (#11) 2022-12-16 16:03:39 +01:00