| .. |
|
__init__.py
|
ece539abd6
Add __init__.py files to subdirectories for installation
|
il y a 2 ans |
|
baichuan.py
|
3f7d5786ba
Pass alibi slopes to flash_attn_with_kvcache during generation
|
il y a 11 mois |
|
bert.py
|
30e1ef0f79
minify torch.torch.int32 to torch.int32 (#1237)
|
il y a 2 mois |
|
bigcode.py
|
07005806ff
Add BigCode converters (#532)
|
il y a 1 an |
|
btlm.py
|
7ffba9a501
Implement BTLM model
|
il y a 11 mois |
|
falcon.py
|
f1a73d0740
Run isort and black on python files
|
il y a 1 an |
|
gpt.py
|
0d810cfb73
Fix KeyError handling for non-existing key in state_dict.pop() (#898)
|
il y a 5 mois |
|
gpt_neox.py
|
0a146185d6
[Gen] Remove minor dead code
|
il y a 1 an |
|
gptj.py
|
f1a73d0740
Run isort and black on python files
|
il y a 1 an |
|
llama.py
|
187c2a0635
Fix E1136 (#563)
|
il y a 1 an |
|
opt.py
|
f1a73d0740
Run isort and black on python files
|
il y a 1 an |
|
vit.py
|
abbc131173
[LayerNorm] Switch from CUDA to Triton implementation
|
il y a 11 mois |
Zhihao Shen
30e1ef0f79
minify torch.torch.int32 to torch.int32 (#1237)