 Guillaume Wenzek
|
eb7810b81f
force little-endian
|
hai 1 ano |
 Ning
|
d9061b89b3
Update test_unity_cpp.py
|
hai 1 ano |
|
Ning
|
1589c529bf
unity.cpp speech_encoder_frontend+speech_encoder+adaptor (#81)
|
hai 1 ano |
|
Guillaume Wenzek
|
e38442d1f0
rename mask to attn_mask in MultiheadAttention_forward
|
hai 1 ano |
|
Guillaume Wenzek
|
6fbb465f2b
generate_sequence return full results
|
hai 1 ano |
|
Guillaume Wenzek
|
f49763de86
fix beam size and scores
|
hai 1 ano |
|
Guillaume Wenzek
|
c28db8c8ac
wip beam-size=2
|
hai 1 ano |
|
Guillaume Wenzek
|
3d69d4975f
batching -> StandardTransformerDecoder
|
hai 1 ano |
|
Guillaume Wenzek
|
b24dbe3030
batching -> TransformerEmbeddingFrontend_forward
|
hai 1 ano |
|
Guillaume Wenzek
|
86993cbd00
fix StandardTransformerEncoder
|
hai 1 ano |
|
Guillaume Wenzek
|
28ed039370
fix MultiheadAttention_forward
|
hai 1 ano |
|
Guillaume Wenzek
|
81cdf80eb9
WIP: MultiheadAttention_forward
|
hai 1 ano |
|
Guillaume Wenzek
|
f2b5007277
start testing with batch size > 1
|
hai 1 ano |
|
Guillaume Wenzek
|
eb80195345
use ggml_diag_mask_inf
|
hai 1 ano |
|
Guillaume Wenzek
|
88b0690a72
split tests files
|
hai 1 ano |
|
Guillaume Wenzek
|
bfbafd9603
fix generation with beam_size=1
|
hai 1 ano |
|
Guillaume Wenzek
|
45f986055a
add naive tweaking of lprobs
|
hai 1 ano |
|
Guillaume Wenzek
|
7c9b2a1b95
pass correct prefix sequence in test
|
hai 1 ano |
|
Guillaume Wenzek
|
dcb9535666
wip: generate_sequence
|
hai 1 ano |
|
Guillaume Wenzek
|
2238cea072
SinusoidalPositionEncoder + WIP: TransformerEmbeddingFrontend
|
hai 1 ano |
|
Guillaume Wenzek
|
f1f33dbec1
has_layer + transformer decoder
|
hai 1 ano |
|
Guillaume Wenzek
|
9b157d0384
allclose
|
hai 1 ano |
|
Guillaume Wenzek
|
e1faba0c5d
text encoder
|
hai 1 ano |
|
Guillaume Wenzek
|
b81061704c
StandardTransformerEncoderLayer_forward
|
hai 1 ano |
|
Guillaume Wenzek
|
e7c3b7a4ba
allow flash attn
|
hai 1 ano |
|
Guillaume Wenzek
|
b07a08102a
MultiheadAttention_forward
|
hai 1 ano |
|
Guillaume Wenzek
|
3f1d6992f3
fix to_numpy for transposed tensors
|
hai 1 ano |
|
Guillaume Wenzek
|
fa85f05545
test flash_attn
|
%!s(int64=2) %!d(string=hai) anos |
|
Guillaume Wenzek
|
06d4ed1475
ggml.ne != np.shape
|
%!s(int64=2) %!d(string=hai) anos |
|
Guillaume Wenzek
|
c2e6384e29
nb
|
%!s(int64=2) %!d(string=hai) anos |
