support FlashAttention2

Browse Source

Former-commit-id: 23e56c5554b948d4f08ad87849b261eafd2c7890

...

This commit is contained in:

hiyouga

2023-09-10 20:43:56 +08:00

parent b481ad58e6

commit a402161631

9 changed files with 875 additions and 115 deletions

Show all changes Ignore whitespace when comparing lines Ignore changes in amount of whitespace Ignore changes in whitespace at EOL

Download Patch File Download Diff File Expand all files Collapse all files

0

src/llmtuner/extras/models/__init__.py Normal file