outlines icon indicating copy to clipboard operation
outlines copied to clipboard

Update logits array in-place

Open brandonwillard opened this issue 2 months ago • 0 comments

What behavior of the library made you think about the improvement?

The current structured generation code is creating a -inf copy of the logits array and setting the allowed token ID indices to the corresponding values in the original logits array. See here.

How would you like it to behave?

When possible, the original logits array should be updated in-place and completely avoid creating a new array. This change would likely require the set of disallowed token IDs instead of the allowed ones.

brandonwillard avatar May 02 '24 16:05 brandonwillard