huggingface/text-generation-inference
5
0
Fork 0
mirror of https://github.com/huggingface/text-generation-inference.git synced 2025-04-21 23:12:07 +00:00
Code Issues Packages Projects Releases Wiki Activity
f171bdc823
text-generation-inference/server/text_generation_server/models
History
SeongBeomLEE 66914f7b19
fix: LlamaTokenizerFast to AutoTokenizer at flash_mistral.py (#1637) 
# What does this PR do?

A few cases where you're using a mistral structure or mixtral structure
but not a llama tokenizer, why not make it to call the AutoTokenizer in
exception handling.

Similar PR #619

@Narsil
2024-03-22 17:13:13 +01:00
..
custom_modeling feat: support force downcast after FastRMSNorm multiply for Gemma (#1658) 2024-03-21 10:25:11 +01:00
__init__.py feat: accept legacy request format and response (#1527) 2024-02-29 10:44:20 -05:00
bloom.py Revamp medusa implementation so that every model can benefit. (#1588) 2024-02-26 19:49:28 +01:00
cache_manager.py feat: add mistral model (#1071) 2023-09-28 09:55:47 +02:00
causal_lm.py feat: Qwen2 (#1608) 2024-02-28 15:50:31 +01:00
flash_causal_lm.py Remove unecessary cuda graph. (#1664) 2024-03-21 20:11:05 +01:00
flash_gemma.py Revamp medusa implementation so that every model can benefit. (#1588) 2024-02-26 19:49:28 +01:00
flash_llama.py Revamp medusa implementation so that every model can benefit. (#1588) 2024-02-26 19:49:28 +01:00
flash_mistral.py fix: LlamaTokenizerFast to AutoTokenizer at flash_mistral.py (#1637) 2024-03-22 17:13:13 +01:00
flash_mixtral.py Revamp medusa implementation so that every model can benefit. (#1588) 2024-02-26 19:49:28 +01:00
flash_neox.py Revamp medusa implementation so that every model can benefit. (#1588) 2024-02-26 19:49:28 +01:00
flash_phi.py Revamp medusa implementation so that every model can benefit. (#1588) 2024-02-26 19:49:28 +01:00
flash_qwen2.py feat: Qwen2 (#1608) 2024-02-28 15:50:31 +01:00
flash_rw.py Revamp medusa implementation so that every model can benefit. (#1588) 2024-02-26 19:49:28 +01:00
flash_santacoder.py Revamp medusa implementation so that every model can benefit. (#1588) 2024-02-26 19:49:28 +01:00
flash_starcoder2.py feat: Qwen2 (#1608) 2024-02-28 15:50:31 +01:00
galactica.py feat: Qwen2 (#1608) 2024-02-28 15:50:31 +01:00
globals.py Small cleanup. (#1560) 2024-02-14 15:30:07 +01:00
gpt_neox.py feat: Qwen2 (#1608) 2024-02-28 15:50:31 +01:00
idefics_causal_lm.py Revamp medusa implementation so that every model can benefit. (#1588) 2024-02-26 19:49:28 +01:00
idefics.py Fix idefics default. (#1614) 2024-02-29 13:16:34 +01:00
mamba.py Revamp medusa implementation so that every model can benefit. (#1588) 2024-02-26 19:49:28 +01:00
model.py fix: fix logic if sliding window key is not present in config (#1352) 2023-12-15 14:56:17 +01:00
mpt.py Revamp medusa implementation so that every model can benefit. (#1588) 2024-02-26 19:49:28 +01:00
opt.py Revamp medusa implementation so that every model can benefit. (#1588) 2024-02-26 19:49:28 +01:00
phi.py Revamp medusa implementation so that every model can benefit. (#1588) 2024-02-26 19:49:28 +01:00
rw.py feat: Qwen2 (#1608) 2024-02-28 15:50:31 +01:00
santacoder.py Revamp medusa implementation so that every model can benefit. (#1588) 2024-02-26 19:49:28 +01:00
seq2seq_lm.py feat: Qwen2 (#1608) 2024-02-28 15:50:31 +01:00
t5.py Revamp medusa implementation so that every model can benefit. (#1588) 2024-02-26 19:49:28 +01:00
types.py chore: add pre-commit (#1569) 2024-02-16 11:58:58 +01:00