Skip to content

Commit c99a2d9

Browse files
committed
llama : add comments about experimental flags
1 parent b9adcbb commit c99a2d9

File tree

1 file changed

+4
-3
lines changed

1 file changed

+4
-3
lines changed

llama.h

Lines changed: 4 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -264,6 +264,7 @@ extern "C" {
264264
bool check_tensors; // validate model tensor data
265265
};
266266

267+
// NOTE: changing the default values of parameters marked as [EXPERIMENTAL] may cause crashes or incorrect results in certain configurations
267268
struct llama_context_params {
268269
uint32_t seed; // RNG seed, -1 for random
269270
uint32_t n_ctx; // text context, 0 = from model
@@ -290,14 +291,14 @@ extern "C" {
290291
ggml_backend_sched_eval_callback cb_eval;
291292
void * cb_eval_user_data;
292293

293-
enum ggml_type type_k; // data type for K cache
294-
enum ggml_type type_v; // data type for V cache
294+
enum ggml_type type_k; // data type for K cache [EXPERIMENTAL]
295+
enum ggml_type type_v; // data type for V cache [EXPERIMENTAL]
295296

296297
// Keep the booleans together to avoid misalignment during copy-by-value.
297298
bool logits_all; // the llama_decode() call computes all logits, not just the last one (DEPRECATED - set llama_batch.logits instead)
298299
bool embeddings; // if true, extract embeddings (together with logits)
299300
bool offload_kqv; // whether to offload the KQV ops (including the KV cache) to GPU
300-
bool flash_attn; // whether to use flash attention
301+
bool flash_attn; // whether to use flash attention [EXPERIMENTAL]
301302

302303
// Abort callback
303304
// if it returns true, execution of llama_decode() will be aborted

0 commit comments

Comments
 (0)