|
layers
|
Run isort and black on test files
|
2023-08-18 20:59:35 -07:00 |
|
losses
|
Run isort and black on test files
|
2023-08-18 20:59:35 -07:00 |
|
models
|
Support cache_seqlens being integer
|
2023-09-05 11:27:48 -07:00 |
|
modules
|
Run isort and black on test files
|
2023-08-18 20:59:35 -07:00 |
|
ops
|
Run isort and black on test files
|
2023-08-18 20:59:35 -07:00 |
|
test_flash_attn.py
|
Implement flash_attn_with_kvcache
|
2023-09-04 00:11:44 -07:00 |
|
test_rotary.py
|
[Rotary] Implement varlen rotary
|
2023-09-03 17:57:10 -07:00 |
