Commits · 8a733cbd538ef59aacfd60bc44a5262dbe8f5768 · gaoqiong / flash-attention

11 Sep, 2023 1 commit
- [Gen] Fix calling update_graph_cache in tests · 8a733cbd
  Tri Dao authored Sep 10, 2023
  
  8a733cbd
05 Sep, 2023 1 commit
- [Gen] Refactor decoding function · 913922ca
  Tri Dao authored Sep 04, 2023
  
  913922ca
26 Aug, 2023 1 commit
- [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
- [Gen] Add OPT to generation test · f68d41ec
  Tri Dao authored Jan 17, 2023
  
  f68d41ec
16 Jan, 2023 2 commits
- [ViT] Support timm checkpoint, add tests · 780e8eea
  Tri Dao authored Jan 16, 2023
  
  780e8eea
- Reorder LN in Block, support OPT · ff34123b
  Tri Dao authored Jan 15, 2023
  
  ff34123b