|
|
1a7eb7da61
|
Support beam search & parallel generation (#7)
|
2023-03-10 09:58:21 -08:00 |
|
|
|
0deacbce6e
|
Implement single_query_cached_kv_attention kernel (#3)
|
2023-03-01 15:02:19 -08:00 |
|
|
|
1ce1333573
|
Set default dtype to half
|
2023-02-23 21:31:39 +00:00 |
|
|
|
fdd0f2f472
|
Minor
|
2023-02-23 20:23:47 +00:00 |
|
|
|
343cea3dbc
|
Add seq_ids to input metadata
|
2023-02-23 09:25:01 +00:00 |
|
|
|
4b1ac23f53
|
Fix slot mapping
|
2023-02-23 00:10:07 +00:00 |
|
|
|
8290fce47d
|
Add Worker class
|
2023-02-22 19:01:38 +00:00 |
|
