Update personality.py

ParisNeo · Jul 29, 2024 · 199c9e2 · 199c9e2
1 parent df7c88e
commit 199c9e2
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/lollms/personality.py b/lollms/personality.py
@@ -700,7 +700,7 @@ def fast_gen(
         prompt = pr.build(placeholders,
                         self.model.tokenize,
                         self.model.detokenize,
-                        (self.lollms.model.config.ctx_size - max_generation_size) if max_generation_size is not None else (self.lollms.model.config.ctx_size - 1024),
+                        (self.lollms.model.config.ctx_size - max_generation_size) if max_generation_size is not None else (self.model.config.ctx_size - 1024),
                         sacrifice
                         )
         # TODO : add show progress