huggingface/text-generation-inference
5
0
Fork 0
mirror of https://github.com/huggingface/text-generation-inference.git synced 2025-10-10 23:45:23 +00:00
Code Issues Packages Projects Releases Wiki Activity
a172430d8b
text-generation-inference/server/text_generation/models
History
Nick Hill a172430d8b fix: Some small fixes 
- Avoid theoretical hang in batcher loop
- Avoid a couple of clones in py server generate method
- Keep attention mask tensors as integers
2022-12-01 16:19:12 -08:00
..
__init__.py feat(server): Support Galactica (#4) 2022-12-01 19:31:54 +01:00
bloom.py feat(server): Support Galactica (#4) 2022-12-01 19:31:54 +01:00
causal_lm.py fix: Some small fixes 2022-12-01 16:19:12 -08:00
galactica.py feat(server): Support Galactica (#4) 2022-12-01 19:31:54 +01:00
model.py feat(server): Support AutoModelForSeq2SeqLM 2022-11-04 18:03:04 +01:00
seq2seq_lm.py fix: Some small fixes 2022-12-01 16:19:12 -08:00
types.py feat(server): Support AutoModelForSeq2SeqLM 2022-11-04 18:03:04 +01:00