Skip to content

Commit da02eae

Browse files
committed
transformerless_lm: add unpronounceable_mask to _single_stage_refine sig
1 parent 5a1fe73 commit da02eae

1 file changed

Lines changed: 2 additions & 1 deletion

File tree

experiments/transformerless_lm/train_self_recursive.py

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1410,7 +1410,8 @@ def _single_stage_refine(model, draft, vocab_size, scorer, mode: str,
14101410
vowel_start_mask: torch.Tensor = None,
14111411
end_vowels: list = None,
14121412
punct_mask: torch.Tensor = None,
1413-
newline_mask: torch.Tensor = None):
1413+
newline_mask: torch.Tensor = None,
1414+
unpronounceable_mask: torch.Tensor = None):
14141415
"""One refinement stage: optimize a single score until plateau.
14151416
14161417
mode: 'min' (harmony, quality) or 'max' (creativity).

0 commit comments

Comments
 (0)