Add Parakeet-RNNT #46331
eustlb
merged
52 commits into
main
from
parakeet-rnnt-from-scratch
add ParakeetForRNNT
e03fdbb6
ParakeetForRNNT config
328c94e3
generation mixin for RNNT
178a33b3
conversion script update
f406106b
auto mappings
38d72d87
tests
1efefcfc
RNNT as the main class
aee4bb7b
rnn-t as main config
29f75424
nit
a90b87d4
update test with reproducers
f38f7510
add RNN-T loss
a1c0785b
fix generate
b7e43d07
correct loss function usage
299f1494
processing handle ctc/rnnt/tdt diffs
23a01a99
udpate doc + fix pipeline
4bdf5c8b
update loss reduction to match NeMo
8c3e43ae
udpate expected value
f4f2c771
loss reduction
a67da591
loss reduction to tdt loss
ddcb571e
conversion udpate
182f4d53
fix
9b3ecc73
update checkpoint
a8e121b6
Merge branch 'main' into parakeet-rnnt-from-scratch
bc77d82b
nit
31a1ea7f
nit
3ae2ee49
fix tdt loss
9d6e6a69
fix loss
309fe6e2
fix loss test
99650000
make
6d041fb1
use correct checkpoint
cdfb7cca
Merge branch 'main' into parakeet-rnnt-from-scratch
bfb2eec6
AutoModelForRNNT in auto.md
d5b552ac
Merge branch 'parakeet-rnnt-from-scratch' of github.com:huggingface/t…
5d140793
fix
7632aa11
Merge branch 'main' into parakeet-rnnt-from-scratch
e98c3dcd
add revision
4207b26f
Merge branch 'parakeet-rnnt-from-scratch' of github.com:huggingface/t…
b784e393
Update src/transformers/loss/loss_rnnt.py
0dac0062
address review
ecb8659d
Update src/transformers/loss/loss_rnnt.py
b33dde65
hf paper alternative
cac8e1bf
back to mean default for tdt loss
d117b68d
re-add todo
2c295d46
ensure bc
a60a1411
add FIXTURES_DIR
965c04b3
cleaner approach
bc3500dd
loss reduction as a kwarg
c80e304e
revert reduction="mean" for BC in tdt loss
3f35faa6
add RNNTLossTest
db82fd6c
vasqu
approved these changes
on 2026-06-11
Merge branch 'main' into parakeet-rnnt-from-scratch
ce3e6028
Merge branch 'main' into parakeet-rnnt-from-scratch
398fbd1b
revert to inheritance on __init__
d03568e3
Login to write a write a comment.
Login via GitHub