diff --git a/characters/instruction-following/Koala.yaml b/characters/instruction-following/Koala.yaml new file mode 100644 index 00000000..18dc7b9b --- /dev/null +++ b/characters/instruction-following/Koala.yaml @@ -0,0 +1,3 @@ +name: "GPT:" +your_name: "USER:" +context: "BEGINNING OF CONVERSATION:" diff --git a/models/config.yaml b/models/config.yaml index 7701ecc4..2cb09ed0 100644 --- a/models/config.yaml +++ b/models/config.yaml @@ -42,3 +42,6 @@ llama-[0-9]*b-4bit$: instruction_template: 'Alpaca' skip_special_tokens: false custom_stopping_strings: '"### End"' +.*koala: + mode: 'instruct' + instruction_template: 'Koala'