|
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
|
[Gen] Clone logits before returning when cg=True
|
2023-08-27 23:19:58 -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 splitKV attention
|
2023-08-29 00:58:29 -07:00 |
|
test_rotary.py
|
Run isort and black on test files
|
2023-08-18 20:59:35 -07:00 |
