From cb95a2432c9506e494e018f2424f04b934bb678c Mon Sep 17 00:00:00 2001
From: oobabooga <112222186+oobabooga@users.noreply.github.com>
Date: Sun, 16 Apr 2023 14:40:45 -0300
Subject: [PATCH] Add Koala support

---
 characters/instruction-following/Koala.yaml | 3 +++
 models/config.yaml                          | 3 +++
 2 files changed, 6 insertions(+)
 create mode 100644 characters/instruction-following/Koala.yaml

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'