From a8ae73d5002d46c7ea1abb59fcac7912061b2256 Mon Sep 17 00:00:00 2001 From: Saifeddine ALOUI Date: Mon, 29 Jul 2024 16:38:37 +0200 Subject: [PATCH] Update personality.py --- lollms/personality.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/lollms/personality.py b/lollms/personality.py index f78f1b7..61ece57 100644 --- a/lollms/personality.py +++ b/lollms/personality.py @@ -700,7 +700,7 @@ class AIPersonality: 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.model.config.ctx_size - 1024), + (self.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