diff --git a/matrix_pygmalion_bot/ai/model_helpers.py b/matrix_pygmalion_bot/ai/model_helpers.py index 5b9e2d0..75e136b 100644 --- a/matrix_pygmalion_bot/ai/model_helpers.py +++ b/matrix_pygmalion_bot/ai/model_helpers.py @@ -65,8 +65,8 @@ async def get_full_prompt(simple_prompt: str, bot, chat_history, model_name: str elif model_name.startswith("alpaca"): prompt = f"Below is an instruction that describes a task, paired with an input that provides further context. Write a response that appropriately completes the request.\n\n" prompt += f"### Instruction:\nGiven the following character description and scenario, write a script for a dialogue between the human user {bot.user_name} and the fictional AI assistant {bot.name}. Play the role of the character {bot.name}.\n" - if bot.nsfw: - prompt += f"NSFW content allowed.\n" # ToDo +# if bot.nsfw: +# prompt += f"NSFW content allowed.\n" # ToDo prompt += f"\n" prompt += "### Input:\n" elif model_name.startswith("koboldai"):