| .. |
|
custom_modeling
|
feat: add support for Gemma (#1583)
|
2024-02-21 14:15:22 +01:00 |
|
__init__.py
|
feat: add support for Gemma (#1583)
|
2024-02-21 14:15:22 +01:00 |
|
bloom.py
|
fix: fix gpt-q params loading
|
2023-12-14 11:02:16 +01:00 |
|
cache_manager.py
|
feat: add mistral model (#1071)
|
2023-09-28 09:55:47 +02:00 |
|
causal_lm.py
|
Outlines guided generation (#1539)
|
2024-02-15 10:28:10 +01:00 |
|
flash_causal_lm.py
|
chore: add pre-commit (#1569)
|
2024-02-16 11:58:58 +01:00 |
|
flash_gemma.py
|
feat: add support for Gemma (#1583)
|
2024-02-21 14:15:22 +01:00 |
|
flash_llama.py
|
v1.4.0 (#1494)
|
2024-01-26 19:04:57 +01:00 |
|
flash_mistral.py
|
Fix mistral with length > window_size for long prefills (rotary doesn't create long enough cos, sin). (#1571)
|
2024-02-19 15:23:12 +01:00 |
|
flash_mixtral.py
|
chore: formatting
|
2023-12-11 14:49:52 +01:00 |
|
flash_neox.py
|
fix: fix gpt-q params loading
|
2023-12-14 11:02:16 +01:00 |
|
flash_phi.py
|
v1.4.0 (#1494)
|
2024-01-26 19:04:57 +01:00 |
|
flash_rw.py
|
fix: fix gpt-q params loading
|
2023-12-14 11:02:16 +01:00 |
|
flash_santacoder.py
|
fix: fix gpt-q params loading
|
2023-12-14 11:02:16 +01:00 |
|
galactica.py
|
chore: add pre-commit (#1569)
|
2024-02-16 11:58:58 +01:00 |
|
globals.py
|
Small cleanup. (#1560)
|
2024-02-14 15:30:07 +01:00 |
|
gpt_neox.py
|
fix: fix gpt-q params loading
|
2023-12-14 11:02:16 +01:00 |
|
idefics.py
|
enable bfloat16 for cpu (#1034)
|
2023-09-19 17:19:28 +02:00 |
|
idefics_causal_lm.py
|
chore: add pre-commit (#1569)
|
2024-02-16 11:58:58 +01:00 |
|
mamba.py
|
chore: add pre-commit (#1569)
|
2024-02-16 11:58:58 +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
|
fix: fix gpt-q params loading
|
2023-12-14 11:02:16 +01:00 |
|
opt.py
|
fix: fix gpt-q params loading
|
2023-12-14 11:02:16 +01:00 |
|
phi.py
|
v1.4.0 (#1494)
|
2024-01-26 19:04:57 +01:00 |
|
rw.py
|
chore: add pre-commit (#1569)
|
2024-02-16 11:58:58 +01:00 |
|
santacoder.py
|
enable bfloat16 for cpu (#1034)
|
2023-09-19 17:19:28 +02:00 |
|
seq2seq_lm.py
|
chore: add pre-commit (#1569)
|
2024-02-16 11:58:58 +01:00 |
|
t5.py
|
enable bfloat16 for cpu (#1034)
|
2023-09-19 17:19:28 +02:00 |
|
types.py
|
chore: add pre-commit (#1569)
|
2024-02-16 11:58:58 +01:00 |
c86f58d37c