Further limit the number of tokens

app.py

@@ -111,6 +111,7 @@ window_len = st.select_slider(
 # Now figure out how many tokens we are allowed to use:
 # window_len * (num_tokens + window_len) * vocab_size <= MAX_MEM
 max_tokens = int(MAX_MEM / (multiplier * window_len) - window_len)
+max_tokens = min(max_tokens, 4096)
 
 DEFAULT_TEXT = """
 We present context length probing, a novel explanation technique for causal