From 563a79afa1c6a5ad75ab88aef805f79fb1559f73 Mon Sep 17 00:00:00 2001
From: Daniel Clough <9276072+danielclough@users.noreply.github.com>
Date: Wed, 20 Dec 2023 17:16:31 -0800
Subject: [PATCH] make fn name generic (#1459)

Co-authored-by: Ubuntu <danielclough@users.noreply.github.com>
---
 candle-transformers/src/models/mistral.rs | 3 ++-
 1 file changed, 2 insertions(+), 1 deletion(-)

diff --git a/candle-transformers/src/models/mistral.rs b/candle-transformers/src/models/mistral.rs
index 0ba556bd..fe3187b9 100644
--- a/candle-transformers/src/models/mistral.rs
+++ b/candle-transformers/src/models/mistral.rs
@@ -40,7 +40,8 @@ impl Config {
     }
 
     // https://huggingface.co/Open-Orca/Mistral-7B-OpenOrca/blob/main/config.json
-    pub fn config_open_orca_chat_ml(use_flash_attn: bool) -> Self {
+    // https://huggingface.co/teknium/OpenHermes-2.5-Mistral-7B/blob/main/config.json
+    pub fn config_chat_ml(use_flash_attn: bool) -> Self {
         Self {
             vocab_size: 32002,
             hidden_size: 4096,