diff --git a/src/lmql/tests/test_stopping.py b/src/lmql/tests/test_stopping.py index 82b15dff..007d6988 100644 --- 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()) \ No newline at end of file + run_all_tests(globals())