fix stopping test

eth-sri · Feb 14, 2024 · a9bd191 · a9bd191
1 parent b6d2dc1
commit a9bd191
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/src/lmql/tests/test_stopping.py b/src/lmql/tests/test_stopping.py
@@ -64,7 +64,7 @@ async def test_conditional_stopping():
     from 
         lmql.model("local:llama.cpp:/lmql/llama-2-7b-chat.Q2_K.gguf", tokenizer="AyyYOO/Luna-AI-Llama2-Uncensored-FP16-sharded") 
     where 
-        len(TOKENS(OUTPUT)) > 21 and STOPS_AT(OUTPUT, "review")
+        len(TOKENS(OUTPUT)) > 80 and STOPS_AT(OUTPUT, "review")
     '''
 
 @lmql.query