From 06c7db017da935d7db3341124e6d61d5828703ed Mon Sep 17 00:00:00 2001 From: Matthew McAllister Date: Tue, 9 May 2023 16:31:27 -0700 Subject: [PATCH] Add config for pygmalion-7b and metharme-7b (#1887) --- models/config.yaml | 4 ++++ 1 file changed, 4 insertions(+) diff --git a/models/config.yaml b/models/config.yaml index 7690d527..ebc92898 100644 --- a/models/config.yaml +++ b/models/config.yaml @@ -59,3 +59,7 @@ llama-[0-9]*b-4bit$: .*moss-moon.*sft: mode: 'instruct' instruction_template: 'MOSS' +.*pygmalion-7b: + model_type: 'llama' +.*metharme-7b: + model_type: 'llama'