Skip to content

Commit 91081e7

Browse files
committed
no heap alloc
1 parent 930dc9a commit 91081e7

File tree

1 file changed

+3
-2
lines changed

1 file changed

+3
-2
lines changed

simple/src/main.rs

Lines changed: 3 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -22,6 +22,7 @@ use llama_cpp_2::token::data_array::LlamaTokenDataArray;
2222
use std::io::Write;
2323
use std::num::NonZeroU32;
2424
use std::path::PathBuf;
25+
use std::pin::pin;
2526
use std::str::FromStr;
2627
use std::time::Duration;
2728
use llama_cpp_2::model::params::kv_overrides::ParamOverrideValue;
@@ -120,11 +121,11 @@ fn main() -> Result<()> {
120121
LlamaModelParams::default()
121122
};
122123

123-
let mut model_params = Box::pin(model_params);
124+
let mut model_params = pin!(model_params);
124125

125126
for (k, v) in key_value_overrides.iter() {
126127
let k = CString::new(k.as_bytes()).with_context(|| format!("invalid key: {}", k))?;
127-
model_params.as_mut().append_kv_override(k.as_c_str(), *v);
128+
model_params.append_kv_override(k.as_c_str(), *v);
128129
}
129130

130131
let model_path = model

0 commit comments

Comments
 (0)