Skip to content
GitLab
Menu
Projects
Groups
Snippets
Loading...
Help
Help
Support
Community forum
Keyboard shortcuts
?
Submit feedback
Contribute to GitLab
Sign in / Register
Toggle navigation
Menu
Open sidebar
gaoqiong
flash-attention
Commits
f8aea6ead01028599c8f47a56691dad3828935c0
Switch branch/tag
flash-attention
tests
models
test_gpt.py
27 Aug, 2023
1 commit
[GPT] Generalize last_token_only arg to num_last_tokens
· f8aea6ea
Tri Dao
authored
Aug 26, 2023
f8aea6ea
26 Aug, 2023
3 commits
[GPT] Test generation when passing in multiple tokens
· 371e2065
Tri Dao
authored
Aug 26, 2023
371e2065
[GPT] Move more tests to test_gpt.py
· c000c3a2
Tri Dao
authored
Aug 26, 2023
c000c3a2
[GPT] Move GPT and OPT generation tests to test_{gpt,opt}.py
· 9b713872
Tri Dao
authored
Aug 26, 2023
9b713872
19 Aug, 2023
1 commit
Run isort and black on test files
· 0e8c46ae
Tri Dao
authored
Aug 18, 2023
0e8c46ae
22 Mar, 2023
1 commit
Implement GPT-J
· 4d87e4d8
Tri Dao
authored
Mar 22, 2023
4d87e4d8
18 Jan, 2023
1 commit
[FusedDense] Support relu, rename FusedDenseGeluDense -> FusedMLP
· 88173a1a
Tri Dao
authored
Jan 17, 2023
88173a1a
16 Jan, 2023
1 commit
Reorder LN in Block, support OPT
· ff34123b
Tri Dao
authored
Jan 15, 2023
ff34123b
28 Dec, 2022
1 commit
Implement generation for GPT
· 63670fd8
Tri Dao
authored
Dec 27, 2022
63670fd8
27 Dec, 2022
1 commit
Support loading GPT2 weights from Huggingface
· 9d797d88
Tri Dao
authored
Dec 27, 2022
9d797d88
