From efc36c9acfe80be4879c5e1a1954f8053cb3cf53 Mon Sep 17 00:00:00 2001 From: Michal Moskal Date: Sun, 26 Jan 2025 10:15:22 -0800 Subject: [PATCH] add $LLGUIDANCE_LOG_LEVEL support --- common/llguidance.cpp | 5 ++++- 1 file changed, 4 insertions(+), 1 deletion(-) diff --git a/common/llguidance.cpp b/common/llguidance.cpp index 17fa52752..31204c20d 100644 --- a/common/llguidance.cpp +++ b/common/llguidance.cpp @@ -21,7 +21,10 @@ static LlgConstraint * llama_sampler_llg_new(LlgTokenizer * tokenizer, const cha const char * grammar_data) { LlgConstraintInit cinit; llg_constraint_init_set_defaults(&cinit, tokenizer); - // cinit.log_stderr_level = 2; + const char *log_level = getenv("LLGUIDANCE_LOG_LEVEL"); + if (log_level && *log_level) { + cinit.log_stderr_level = atoi(log_level); + } auto c = llg_new_constraint_any(&cinit, grammar_kind, grammar_data); if (llg_get_error(c)) { LOG_ERR("llg error: %s\n", llg_get_error(c));