Woosuk Kwon
|
1a7eb7da61
|
Support beam search & parallel generation (#7)
|
2023-03-10 09:58:21 -08:00 |
|
Woosuk Kwon
|
04e5acc08e
|
Fix a bug in 1D input shape (#5)
|
2023-03-06 10:05:27 -08:00 |
|
Woosuk Kwon
|
3e9f991d6a
|
Use FlashAttention for multi_query_kv_attention (#4)
|
2023-03-01 21:13:08 -08:00 |
|
Woosuk Kwon
|
0deacbce6e
|
Implement single_query_cached_kv_attention kernel (#3)
|
2023-03-01 15:02:19 -08:00 |
|
Woosuk Kwon
|
cbf8779afa
|
Fix a bug in tying OPT embeddings (#1)
|
2023-02-24 16:29:36 -08:00 |
|
Woosuk Kwon
|
762fd1c3fa
|
Refactor and annotate types for attention
|
2023-02-24 08:58:46 +00:00 |
|
Woosuk Kwon
|
7f22f90e8c
|
Remove xformers
|
2023-02-24 08:36:16 +00:00 |
|
Woosuk Kwon
|
932844f1cd
|
Fix attention
|
2023-02-23 23:02:25 +00:00 |
|
Woosuk Kwon
|
ba84b8728a
|
Fix attention
|
2023-02-23 22:29:46 +00:00 |
|
Woosuk Kwon
|
87e0bcd426
|
Fix attention
|
2023-02-23 21:32:02 +00:00 |
|
Woosuk Kwon
|
1ce1333573
|
Set default dtype to half
|
2023-02-23 21:31:39 +00:00 |
|
Woosuk Kwon
|
de0fabbc5c
|
Fix sampler
|
2023-02-23 20:30:12 +00:00 |
|
Woosuk Kwon
|
fdd0f2f472
|
Minor
|
2023-02-23 20:23:47 +00:00 |
|
Woosuk Kwon
|
7f985166f7
|
Consider pempty tensor
|
2023-02-23 20:20:33 +00:00 |
|
Woosuk Kwon
|
86f9eb6d39
|
Fix typo
|
2023-02-23 20:19:24 +00:00 |
|
Woosuk Kwon
|
d4bc1a4d24
|
Add unoptimized OPT Attention
|
2023-02-23 09:31:55 +00:00 |
|
Woosuk Kwon
|
b56b6ca0d6
|
Add greedy sampler
|
2023-02-23 09:26:09 +00:00 |
|
Woosuk Kwon
|
343cea3dbc
|
Add seq_ids to input metadata
|
2023-02-23 09:25:01 +00:00 |
|
Woosuk Kwon
|
4b1ac23f53
|
Fix slot mapping
|
2023-02-23 00:10:07 +00:00 |
|
Woosuk Kwon
|
7b6844e590
|
Add input metadata
|
2023-02-22 19:01:20 +00:00 |
|
Woosuk Kwon
|
608f74ffe5
|
Minor
|
2023-02-22 18:08:25 +00:00 |
|
Woosuk Kwon
|
709a69176e
|
Move worker/models -> models
|
2023-02-22 18:03:48 +00:00 |
|