From 9afb0a1d3296695f44adbe094530a6493287d746 Mon Sep 17 00:00:00 2001 From: Hendrik Langer Date: Fri, 14 Apr 2023 02:05:04 +0200 Subject: [PATCH] use pygmalion again --- matrix_pygmalion_bot/ai/koboldcpp.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/matrix_pygmalion_bot/ai/koboldcpp.py b/matrix_pygmalion_bot/ai/koboldcpp.py index 7a497e6..964b8b8 100644 --- a/matrix_pygmalion_bot/ai/koboldcpp.py +++ b/matrix_pygmalion_bot/ai/koboldcpp.py @@ -12,8 +12,8 @@ import io import base64 from PIL import Image, PngImagePlugin -#from .pygmalion_helpers import get_full_prompt, num_tokens -from .llama_helpers import get_full_prompt, num_tokens +from .pygmalion_helpers import get_full_prompt, num_tokens +#from .llama_helpers import get_full_prompt, num_tokens logger = logging.getLogger(__name__)