.. |
custom_modeling
|
feat: update exllamav2 kernels (#1370)
|
2023-12-21 17:25:22 +01:00 |
__init__.py
|
feat: boilerplate phi2 model integration
|
2024-01-10 09:42:26 -05: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
|
feat: add more latency metrics in forward (#1346)
|
2023-12-14 15:59:38 +01:00 |
flash_causal_lm.py
|
feat: add more latency metrics in forward (#1346)
|
2023-12-14 15:59:38 +01:00 |
flash_llama.py
|
fix: fix gpt-q params loading
|
2023-12-14 11:02:16 +01:00 |
flash_mistral.py
|
fix: fix gpt-q params loading
|
2023-12-14 11:02:16 +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_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
|
fix: fix gpt-q params loading
|
2023-12-14 11:02:16 +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
|
feat: add more latency metrics in forward (#1346)
|
2023-12-14 15:59:38 +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 |
phi2.py
|
feat: boilerplate phi2 model integration
|
2024-01-10 09:42:26 -05:00 |
rw.py
|
enable bfloat16 for cpu (#1034)
|
2023-09-19 17:19:28 +02:00 |
santacoder.py
|
enable bfloat16 for cpu (#1034)
|
2023-09-19 17:19:28 +02:00 |
seq2seq_lm.py
|
feat: add more latency metrics in forward (#1346)
|
2023-12-14 15:59:38 +01:00 |
t5.py
|
enable bfloat16 for cpu (#1034)
|
2023-09-19 17:19:28 +02:00 |
types.py
|
chore: formatting
|
2023-12-11 14:49:52 +01:00 |