Add Koala support

This commit is contained in:
oobabooga 2023-04-16 14:40:45 -03:00
parent b937c9d8c2
commit cb95a2432c
2 changed files with 6 additions and 0 deletions

View File

@ -0,0 +1,3 @@
name: "GPT:"
your_name: "USER:"
context: "BEGINNING OF CONVERSATION:"

View File

@ -42,3 +42,6 @@ llama-[0-9]*b-4bit$:
instruction_template: 'Alpaca' instruction_template: 'Alpaca'
skip_special_tokens: false skip_special_tokens: false
custom_stopping_strings: '"### End"' custom_stopping_strings: '"### End"'
.*koala:
mode: 'instruct'
instruction_template: 'Koala'