 Guillaume Wenzek
|
ddff1a0644
add failing test: test_PositionalEmbedding_forward_with_cache
|
hai 1 ano |
|
Guillaume Wenzek
|
8c074387a9
s2tt generate the sample if needed
|
hai 1 ano |
|
Guillaume Wenzek
|
a771adc782
nicer ctypes_utils
|
hai 1 ano |
|
Guillaume Wenzek
|
f6d810543d
fix incremental decoding !
|
hai 1 ano |
|
Guillaume Wenzek
|
522b97234e
WIP: simple failing test case
|
hai 1 ano |
|
Guillaume Wenzek
|
cc23e2b1c7
prepare tests for flash_attn
|
hai 1 ano |
|
Guillaume Wenzek
|
18c919b0b8
update flash_attn tests
|
hai 1 ano |
|
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 |
