add stacks cache into llama_grammar

This commit is contained in:
Clarissa Miranda 2024-10-17 14:30:07 +11:00
parent 901a3479b1
commit 2aa6dd273a
4 changed files with 15 additions and 9 deletions

View file

@ -34,8 +34,8 @@ static bool match_string(const std::string & input, llama_grammar * grammar) {
const llama_grammar_rules & rules = llama_grammar_get_rules (grammar);
llama_grammar_stacks & stacks_cur = llama_grammar_get_stacks(grammar);
llama_grammar_stacks_cache & stacks_cache = llama_grammar_get_stacks_cache(grammar);
llama_grammar_stacks_cache stacks_cache;
for (const auto & cpt : cpts) {
const llama_grammar_stacks stacks_prev = llama_grammar_get_stacks(grammar); // copy