transformers
Add early stopping for Bark generation via logits processor
#26675
Merged
Merged
Add early stopping for Bark generation via logits processor #26675
gante
merged
29 commits into
huggingface:main
from
isaac-chung:improve-bark-generation
add early stopping logits processor
c621e667
isaac-chung
changed the title
isaac-chung
changed the title
isaac-chung
changed the title
black formmated
9027782f
indent
a93b03cb
follow method signature
1de69b83
actual logic
0937f063
check for None
05d16476
address comments on docstrings and method signature
88571d1e
add unit test under `LogitsProcessorTest` wip
528b9676
unit test passing
dc33e369
black formatted
ce21bc5a
condition per sample
45b3065e
add to BarkModelIntegrationTests
fa5d2516
wip BarkSemanticModelTest
b08346dc
rename and add to kwargs handling
81f90ce5
not add to BarkSemanticModelTest
f9c1c847
correct logic and assert last outputs tokens different in test
a9645bbf
doc-builder style
4c81b7dc
read from kwargs as well
ff7343b6
ArthurZucker
commented
on 2023-10-19
assert len of with less than that of without
9e902320
ruff
a37aac3b
add back seed and test case
b0200b8c
add original impl default suggestion
7c743c64
doc-builder
28c12e92
rename and use softmax
542f41e2
switch back to LogitsProcessor and update docs wording
e16309d8
gante
approved these changes
on 2023-10-25
camelCase and spelling and saving compute
0c75c1c8
assert strictly less than
d88acd82
assert less than
eb9b2f0a
ylacombe
commented
on 2023-10-26
expand test_generate_semantic_early_stop instead
9e835472
Reviewers
Assignees
No one assigned
Labels
None yet
Milestone
No milestone
Login to write a write a comment.
Login via GitHub