Spaces:

pyvene
/

AxBench-ReFT-r1-16K

Runtime error

frankaging commited on Jan 25

Commit

de8f900

1 Parent(s): 7f543e6

final

Files changed (1) hide show

app.py CHANGED Viewed

@@ -90,8 +90,8 @@ def generate(
     max_new_tokens: int=DEFAULT_MAX_NEW_TOKENS,
 ) -> Iterator[str]:
-    # limit to last 3 turns
-    start_idx = max(0, len(chat_history) - 3)
     recent_history = chat_history[start_idx:]
     # build list of messages

     max_new_tokens: int=DEFAULT_MAX_NEW_TOKENS,
 ) -> Iterator[str]:
+    # limit to last 4 turns
+    start_idx = max(0, len(chat_history) - 4)
     recent_history = chat_history[start_idx:]
     # build list of messages