pub struct PenaltyWindow<'a> { /* private fields */ }Expand description
The tokens the penalties may see: a prompt and the tokens generated after it, in that order.
Borrowed rather than owned because this is built once per sampled token on every decode loop in the workspace; an owning window would clone the whole sequence per token.
The two halves are kept separate rather than concatenated because a decode loop already holds them separately, and concatenating would mean an allocation per token for a value only ever read back as “the last N of the two”.
Implementations§
Source§impl<'a> PenaltyWindow<'a>
impl<'a> PenaltyWindow<'a>
Sourcepub fn new(prompt: &'a [usize], generated: &'a [usize]) -> Self
pub fn new(prompt: &'a [usize], generated: &'a [usize]) -> Self
The window over prompt followed by generated.
prompt is the tokens the model was fed before generation
started, and it belongs in the window: see the module docs for
the llama.cpp lines that put it there.
pub fn is_empty(&self) -> bool
Sourcepub fn recent(&self, last_n: usize) -> impl Iterator<Item = usize> + '_
pub fn recent(&self, last_n: usize) -> impl Iterator<Item = usize> + '_
The most recent last_n tokens of prompt ++ generated, oldest
first.
This is llama.cpp’s ring buffer expressed as a view: the buffer
holds at most penalty_last_n entries and the oldest is dropped
on every accept (src/llama-sampler.cpp:2707-2716), so its
contents are exactly the tail of the accepted sequence.
Order does not matter to any caller – only the multiset does – but it is the natural one anyway, and a test reads it.
Trait Implementations§
Source§impl<'a> Clone for PenaltyWindow<'a>
impl<'a> Clone for PenaltyWindow<'a>
Source§fn clone(&self) -> PenaltyWindow<'a>
fn clone(&self) -> PenaltyWindow<'a>
1.0.0 (const: unstable) · Source§fn clone_from(&mut self, source: &Self)
fn clone_from(&mut self, source: &Self)
source. Read moreimpl<'a> Copy for PenaltyWindow<'a>
Auto Trait Implementations§
impl<'a> Freeze for PenaltyWindow<'a>
impl<'a> RefUnwindSafe for PenaltyWindow<'a>
impl<'a> Send for PenaltyWindow<'a>
impl<'a> Sync for PenaltyWindow<'a>
impl<'a> Unpin for PenaltyWindow<'a>
impl<'a> UnsafeUnpin for PenaltyWindow<'a>
impl<'a> UnwindSafe for PenaltyWindow<'a>
Blanket Implementations§
Source§impl<T> BorrowMut<T> for Twhere
T: ?Sized,
impl<T> BorrowMut<T> for Twhere
T: ?Sized,
Source§fn borrow_mut(&mut self) -> &mut T
fn borrow_mut(&mut self) -> &mut T
Source§impl<T> CloneToUninit for Twhere
T: Clone,
impl<T> CloneToUninit for Twhere
T: Clone,
Source§impl<T> IntoEither for T
impl<T> IntoEither for T
Source§fn into_either(self, into_left: bool) -> Either<Self, Self> ⓘ
fn into_either(self, into_left: bool) -> Either<Self, Self> ⓘ
self into a Left variant of Either<Self, Self>
if into_left is true.
Converts self into a Right variant of Either<Self, Self>
otherwise. Read moreSource§fn into_either_with<F>(self, into_left: F) -> Either<Self, Self> ⓘ
fn into_either_with<F>(self, into_left: F) -> Either<Self, Self> ⓘ
self into a Left variant of Either<Self, Self>
if into_left(&self) returns true.
Converts self into a Right variant of Either<Self, Self>
otherwise. Read more