drbh 6f15ac60b2 feat: support force downcast after FastRMSNorm multiply for Gemma (#1658) ... This PR adds `force_downcast_after` to `FastRMSNorm.forward` which is used in the Gemma model. References https://github.com/huggingface/transformers/pull/29402 and https://github.com/huggingface/transformers/pull/29729 Setting `force_downcast_after=True` will perform the `hidden_states * weight` multiplication in f32 and then downcast to half. This differs slightly from the current implementation which first casts the `hidden_states` to a half and then multiples.		2024-03-21 10:25:11 +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	fix: Handle concurrent grammar requests (#1610)	2024-02-29 11:17:42 +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	feat: Qwen2 (#1608)	2024-02-28 15:50:31 +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