Sfoglia il codice sorgente

Add a default prompt for OpenAssistant oasst-sft-1-pythia-12b #253

oobabooga 2 anni fa
parent
commit
0ac562bdba
2 ha cambiato i file con 4 aggiunte e 2 eliminazioni
  1. 2 1
      modules/shared.py
  2. 2 1
      settings-template.json

+ 2 - 1
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|>'
     }
 }
 

+ 2 - 1
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|>"
     }
 }