huggingface/text-generation-inference
5
0
Fork 0
mirror of https://github.com/huggingface/text-generation-inference.git synced 2025-04-22 15:32:08 +00:00
Code Issues Packages Projects Releases Wiki Activity
67ee45a270
text-generation-inference/server/text_generation_server/models
History
yuanwu 67ee45a270 Pass the max_batch_total_tokens to causal_lm 
refine the warmup

Signed-off-by: yuanwu <yuan.wu@intel.com>
2024-10-23 08:28:26 +00:00
..
custom_modeling Make Gaudi adapt to the tgi 2.3.0 2024-09-26 06:04:55 +00:00
__init__.py Make Gaudi adapt to the tgi 2.3.0 2024-09-26 06:04:55 +00:00
bloom.py Make Gaudi adapt to the tgi 2.3.0 2024-09-26 06:04:55 +00:00
causal_lm.py Pass the max_batch_total_tokens to causal_lm 2024-10-23 08:28:26 +00:00
flash_causal_lm.py hotfix : enable intel ipex cpu and xpu in python3.11 (#2517) 2024-09-25 06:15:35 +00:00
galactica.py feat: add ruff and resolve issue (#2262) 2024-09-25 05:46:24 +00:00
globals.py Make Gaudi adapt to the tgi 2.3.0 2024-09-26 06:04:55 +00:00
idefics_causal_lm.py Upgrading exl2. (#2415) 2024-09-25 06:07:40 +00:00
idefics.py Upgrading exl2. (#2415) 2024-09-25 06:07:40 +00:00
mamba.py Fixing exl2 and other quanize tests again. (#2419) 2024-09-25 06:08:38 +00:00
model.py Pass the max_batch_total_tokens to causal_lm 2024-10-23 08:28:26 +00:00
pali_gemma.py feat: add ruff and resolve issue (#2262) 2024-09-25 05:46:24 +00:00
seq2seq_lm.py Fixing exl2 and other quanize tests again. (#2419) 2024-09-25 06:08:38 +00:00
starcoder.py Make Gaudi adapt to the tgi 2.3.0 2024-09-26 06:04:55 +00:00
types.py feat: add ruff and resolve issue (#2262) 2024-09-25 05:46:24 +00:00
vlm_causal_lm.py Make Gaudi adapt to the tgi 2.3.0 2024-09-26 06:04:55 +00:00