Browse Source

instruct

master
Hendrik Langer 2 years ago
parent
commit
a1225413e8
  1. 4
      matrix_pygmalion_bot/ai/model_helpers.py

4
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"): 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"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" 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: # if bot.nsfw:
prompt += f"NSFW content allowed.\n" # ToDo # prompt += f"NSFW content allowed.\n" # ToDo
prompt += f"\n" prompt += f"\n"
prompt += "### Input:\n" prompt += "### Input:\n"
elif model_name.startswith("koboldai"): elif model_name.startswith("koboldai"):

Loading…
Cancel
Save