From 8b1dee3ec89c4904d489455bd7cdab2b0abf1f39 Mon Sep 17 00:00:00 2001 From: oobabooga <112222186+oobabooga@users.noreply.github.com> Date: Wed, 24 Apr 2024 21:47:43 -0700 Subject: [PATCH] Detect platypus-yi-34b, CausalLM-RP-34B, 34b-beta instruction templates --- models/config.yaml | 6 ++++++ 1 file changed, 6 insertions(+) diff --git a/models/config.yaml b/models/config.yaml index 4b9bfd9f..bc12a78f 100644 --- a/models/config.yaml +++ b/models/config.yaml @@ -196,3 +196,9 @@ instruction_template: 'Command-R' .*xwin-lm-70b-v0.1: instruction_template: 'Vicuna-v1.1' +.*platypus-yi-34b: + instruction_template: 'Vicuna-v1.1' +.*CausalLM-RP-34B: + instruction_template: 'ChatML' +34b-beta: + instruction_template: 'ChatML'