Update test_stopping.py

eth-sri · Mar 8, 2024 · 969af9b · 969af9b
1 parent 5c9d802
commit 969af9b
Showing 1 changed file with 2 additions and 2 deletions.
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)) > 25 and STOPS_AT(OUTPUT, "review")
+        len(TOKENS(OUTPUT)) > 80 and STOPS_AT(OUTPUT, "review")
     '''
 
 @lmql.query
@@ -155,4 +155,4 @@ async def test_stop_overlapping_before():
     '''
 
 if __name__ == "__main__":
-    run_all_tests(globals())
+    run_all_tests(globals())