From 0ac562bdba381ed61042802cf56ee3d4e3d674bf Mon Sep 17 00:00:00 2001
From: oobabooga <112222186+oobabooga@users.noreply.github.com>
Date: Sun, 12 Mar 2023 10:46:16 -0300
Subject: [PATCH] Add a default prompt for OpenAssistant oasst-sft-1-pythia-12b
 #253

---
 modules/shared.py      | 3 ++-
 settings-template.json | 3 ++-
 2 files changed, 4 insertions(+), 2 deletions(-)

diff --git a/modules/shared.py b/modules/shared.py
index c42ba7ed..de621710 100644
--- a/modules/shared.py
+++ b/modules/shared.py
@@ -46,7 +46,8 @@ settings = {
     'prompts': {
         'default': 'Common sense questions and answers\n\nQuestion: \nFactual answer:',
         '^(gpt4chan|gpt-4chan|4chan)': '-----\n--- 865467536\nInput text\n--- 865467537\n',
-        '(rosey|chip|joi)_.*_instruct.*': 'User: \n'
+        '(rosey|chip|joi)_.*_instruct.*': 'User: \n',
+        'oasst-*': '<|prompter|>Write a story about future of AI development<|endoftext|><|assistant|>'
     }
 }
 
diff --git a/settings-template.json b/settings-template.json
index 6585f313..9da43970 100644
--- a/settings-template.json
+++ b/settings-template.json
@@ -29,6 +29,7 @@
     "prompts": {
         "default": "Common sense questions and answers\n\nQuestion: \nFactual answer:",
         "^(gpt4chan|gpt-4chan|4chan)": "-----\n--- 865467536\nInput text\n--- 865467537\n",
-        "(rosey|chip|joi)_.*_instruct.*": "User: \n"
+        "(rosey|chip|joi)_.*_instruct.*": "User: \n",
+        "oasst-*": "<|prompter|>Write a story about future of AI development<|endoftext|><|assistant|>"
     }
 }