.. |
layers
|
3ed4cc431c
enc_dec attention code
|
9 months ago |
megatron
|
e42a78381a
feat: switch from pylint to ruff (#322)
|
10 months ago |
models
|
a788ca33bf
hack in custom bias for attention kernels
|
9 months ago |
__init__.py
|
8fa608aeb7
feat: replace Ray with NCCL for control plane comms (#221)
|
11 months ago |
hf_downloader.py
|
e42a78381a
feat: switch from pylint to ruff (#322)
|
10 months ago |
loader.py
|
c2d77b1822
chore: logging refactor (#302)
|
10 months ago |
metadata.py
|
58e89e29d9
add custom bias to attention.py
|
9 months ago |
outlines_decoding.py
|
e42a78381a
feat: switch from pylint to ruff (#322)
|
10 months ago |
outlines_logits_processors.py
|
e42a78381a
feat: switch from pylint to ruff (#322)
|
10 months ago |
sampling_metadata.py
|
9fa99215f8
feat: add cubic sampling (#280)
|
10 months ago |
utils.py
|
2755a48d51
merge dev branch into main (#153)
|
1 year ago |