Skip to content
Merged
Show file tree
Hide file tree
Changes from all commits
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
16 changes: 8 additions & 8 deletions llama-cpp-2/src/sampling.rs
Original file line number Diff line number Diff line change
Expand Up @@ -510,10 +510,6 @@ impl LlamaSampler {
let sampler = unsafe {
llama_cpp_sys_2::llama_sampler_init_dry(
model.vocab_ptr(),
model
.n_ctx_train()
.try_into()
.expect("n_ctx_train exceeds i32::MAX"),
multiplier,
base,
allowed_length,
Expand All @@ -528,20 +524,24 @@ impl LlamaSampler {
/// Penalizes tokens for being present in the context.
///
/// Parameters:
/// - ``penalty_last_n``: last n tokens to penalize (0 = disable penalty, -1 = context size)
/// - ``penalty_repeat``: 1.0 = disabled
/// - ``penalty_freq``: 0.0 = disabled
/// - ``penalty_present``: 0.0 = disabled
/// - ``n_vocab``: [`LlamaModel::n_vocab`]
/// - ``penalty_last_n``: last n tokens to penalize (0 = disable penalty). Negative values are
/// clamped to 0; they no longer select the context size.
/// - ``penalty_repeat``: must be > 0.0, 1.0 = disabled
/// - ``penalty_freq``: must be finite, 0.0 = disabled
/// - ``penalty_present``: must be finite, 0.0 = disabled
#[allow(clippy::too_many_arguments)]
#[must_use]
pub fn penalties(
n_vocab: i32,
penalty_last_n: i32,
penalty_repeat: f32,
penalty_freq: f32,
penalty_present: f32,
) -> Self {
let sampler = unsafe {
llama_cpp_sys_2::llama_sampler_init_penalties(
n_vocab,
penalty_last_n,
penalty_repeat,
penalty_freq,
Expand Down
2 changes: 1 addition & 1 deletion llama-cpp-sys-2/llama.cpp
Submodule llama.cpp updated 1008 files
Loading