From f4b956b47c4c6ec9a2c2a8ea334b12a2f46a5bb8 Mon Sep 17 00:00:00 2001
From: oobabooga <112222186+oobabooga@users.noreply.github.com>
Date: Sat, 25 Nov 2023 06:33:24 -0800
Subject: [PATCH] Detect yi instruction template

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

diff --git a/models/config.yaml b/models/config.yaml
index 703b9e77..f18687ae 100644
--- a/models/config.yaml
+++ b/models/config.yaml
@@ -184,3 +184,5 @@
   instruction_template: 'ChatML'
 .*openhermes.*mistral:
   instruction_template: 'ChatML'
+.*Yi-34B-Chat:
+  instruction_template: 'ChatML'