add pygmalion-2 and mythalion support (#3821)

This commit is contained in:
Eve 2023-09-12 18:57:49 +00:00 committed by GitHub
parent b61d9aef19
commit 90fca6a77d
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23

View File

@ -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' model_type: 'llama'
.*(opt-|opt_|opt1|opt3|optfor|galactica|galpaca|pygmalion-350m): .*(opt-|opt_|opt1|opt3|optfor|galactica|galpaca|pygmalion-350m):
model_type: 'opt' model_type: 'opt'
@ -85,7 +85,7 @@ llama-65b-gptq-3bit:
instruction_template: 'Koala' instruction_template: 'Koala'
.*chatglm: .*chatglm:
instruction_template: 'ChatGLM' instruction_template: 'ChatGLM'
.*metharme: .*(metharme|pygmalion|mythalion):
instruction_template: 'Metharme' instruction_template: 'Metharme'
.*llava: .*llava:
model_type: 'llama' model_type: 'llama'