missing stop token

2023-10-25 01:10:15 -07:00
parent 1e25068bbb
commit 5854307dde
1 changed files with 1 additions and 0 deletions
--- a/memgpt/local_llm/webui/settings.py
+++ b/memgpt/local_llm/webui/settings.py
@@ -2,6 +2,7 @@ SIMPLE = {
    "stopping_strings": [
        "\nUSER:",
        "\nASSISTANT:",
+        "\nFUNCTION RETURN:",
        # '\n' +
        # '</s>',
        # '<|',