AlpinDale a788ca33bf hack in custom bias for attention kernels 9 months ago
..
__init__.py a788ca33bf hack in custom bias for attention kernels 9 months ago
baichuan.py da223153c6 feat&fix: cohere support and missing GPU blocks (#333) 9 months ago
bloom.py da223153c6 feat&fix: cohere support and missing GPU blocks (#333) 9 months ago
chatglm.py e31c6f0b45 feat: refactor modeling logic and support more models (#274) 10 months ago
cohere.py da223153c6 feat&fix: cohere support and missing GPU blocks (#333) 9 months ago
decilm.py e31c6f0b45 feat: refactor modeling logic and support more models (#274) 10 months ago
deepseek.py da223153c6 feat&fix: cohere support and missing GPU blocks (#333) 9 months ago
falcon.py da223153c6 feat&fix: cohere support and missing GPU blocks (#333) 9 months ago
gemma.py da223153c6 feat&fix: cohere support and missing GPU blocks (#333) 9 months ago
gpt2.py da223153c6 feat&fix: cohere support and missing GPU blocks (#333) 9 months ago
gpt_bigcode.py da223153c6 feat&fix: cohere support and missing GPU blocks (#333) 9 months ago
gpt_j.py da223153c6 feat&fix: cohere support and missing GPU blocks (#333) 9 months ago
gpt_neox.py da223153c6 feat&fix: cohere support and missing GPU blocks (#333) 9 months ago
internlm2.py da223153c6 feat&fix: cohere support and missing GPU blocks (#333) 9 months ago
llama.py 6ebac34dc1 chore: cleaner pre-llamafied Yi implementation (#352) 9 months ago
mixtral.py da223153c6 feat&fix: cohere support and missing GPU blocks (#333) 9 months ago
mixtral_quant.py da223153c6 feat&fix: cohere support and missing GPU blocks (#333) 9 months ago
mpt.py da223153c6 feat&fix: cohere support and missing GPU blocks (#333) 9 months ago
olmo.py e42a78381a feat: switch from pylint to ruff (#322) 10 months ago
opt.py da223153c6 feat&fix: cohere support and missing GPU blocks (#333) 9 months ago
phi.py da223153c6 feat&fix: cohere support and missing GPU blocks (#333) 9 months ago
qwen.py da223153c6 feat&fix: cohere support and missing GPU blocks (#333) 9 months ago
qwen2.py da223153c6 feat&fix: cohere support and missing GPU blocks (#333) 9 months ago
stablelm.py da223153c6 feat&fix: cohere support and missing GPU blocks (#333) 9 months ago
t5.py f009f94ffd update modeling code 9 months ago