Skip to content

Commit 2e7b00d

Browse files
committed
Generator: Prevent particular failure mode in generator from holding output indefinitely
1 parent c4867ed commit 2e7b00d

File tree

1 file changed

+3
-1
lines changed

1 file changed

+3
-1
lines changed

exllamav3/generator/job.py

Lines changed: 3 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -583,7 +583,9 @@ def emit(
583583
if not "�" in test_decode:
584584
self.held_text = test_decode
585585
else:
586-
return emit(results)
586+
# Don't hold forever if a broken generation yields a replacement character but never completes
587+
# the Unicode symbol
588+
return emit(results, emit_held = (len(test_decode) > 20))
587589

588590
# Hold text as long as it contains part of a banned string
589591

0 commit comments

Comments
 (0)