hf_text-generation-inference/server/text_generation/models
OlivierDehaene ce960be0a5
feat(bloom): use torch.nn.Linear and torch.nn.GELU (#33)
2023-01-26 15:33:45 +01:00
..
__init__.py feat(server): Support SantaCoder (#26) 2023-01-20 12:24:39 +01:00
bloom.py feat(bloom): use torch.nn.Linear and torch.nn.GELU (#33) 2023-01-26 15:33:45 +01:00
causal_lm.py fix(server): Fix position ids (#28) 2023-01-20 15:35:22 +01:00
galactica.py fix(server): Fix position ids (#28) 2023-01-20 15:35:22 +01:00
model.py fix(server): Minor refactorization using new_zeros (#24) 2023-01-17 09:10:22 +01:00
santacoder.py fix(server): Fix position ids (#28) 2023-01-20 15:35:22 +01:00
seq2seq_lm.py fix(server): Fix position ids (#28) 2023-01-20 15:35:22 +01:00
types.py feat(server): Support SantaCoder (#26) 2023-01-20 12:24:39 +01:00