From 04bd2e408e64d83a8d9c84007707a6b7bf7b5689 Mon Sep 17 00:00:00 2001
From: Jan Ploski <jpl@plosquare.com>
Date: Sat, 17 Jun 2023 04:00:07 +0200
Subject: [PATCH] Fixed segfault during context swap introduced by commit
 3d6ed185

---
 examples/falcon/falcon_main.cpp | 2 ++
 1 file changed, 2 insertions(+)

diff --git a/examples/falcon/falcon_main.cpp b/examples/falcon/falcon_main.cpp
index f0415a65b..d91f50155 100644
--- a/examples/falcon/falcon_main.cpp
+++ b/examples/falcon/falcon_main.cpp
@@ -359,6 +359,8 @@ int main(int argc, char ** argv) {
             if (n_past + (int) embd.size() > n_ctx) {
                 const int n_left = n_past - params.n_keep;
 
+                n_past = params.n_keep;
+
                 // insert n_left/2 tokens at the start of embd from last_n_tokens
                 embd.insert(embd.begin(), last_n_tokens.begin() + n_ctx - n_left/2 - embd.size(), last_n_tokens.end() - embd.size());