From 90fca6a77d416920f2e7439e6fe9e8bdbb6124e9 Mon Sep 17 00:00:00 2001
From: Eve <139727413+netrunnereve@users.noreply.github.com>
Date: Tue, 12 Sep 2023 18:57:49 +0000
Subject: [PATCH] add pygmalion-2 and mythalion support (#3821)

---
 models/config.yaml | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/models/config.yaml b/models/config.yaml
index e69e2562..8783c49b 100644
--- a/models/config.yaml
+++ b/models/config.yaml
@@ -1,4 +1,4 @@
-.*(llama|alpac|vicuna|guanaco|koala|llava|wizardlm|metharme|pygmalion-7b|wizard-mega|openbuddy|vigogne|h2ogpt-research|manticore):
+.*(llama|alpac|vicuna|guanaco|koala|llava|wizardlm|metharme|pygmalion-7b|pygmalion-2|mythalion|wizard-mega|openbuddy|vigogne|h2ogpt-research|manticore):
   model_type: 'llama'
 .*(opt-|opt_|opt1|opt3|optfor|galactica|galpaca|pygmalion-350m):
   model_type: 'opt'
@@ -85,7 +85,7 @@ llama-65b-gptq-3bit:
   instruction_template: 'Koala'
 .*chatglm:
   instruction_template: 'ChatGLM'
-.*metharme:
+.*(metharme|pygmalion|mythalion):
   instruction_template: 'Metharme'
 .*llava:
   model_type: 'llama'