From c0b5c09860923c58660ee9018fa1a5971ff60b76 Mon Sep 17 00:00:00 2001
From: oobabooga <112222186+oobabooga@users.noreply.github.com>
Date: Sat, 22 Apr 2023 15:15:31 -0300
Subject: [PATCH] Minor change

---
 modules/models.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/modules/models.py b/modules/models.py
index ca014d79..469cbaf7 100644
--- a/modules/models.py
+++ b/modules/models.py
@@ -40,7 +40,7 @@ if shared.args.deepspeed:
 
 def find_model_type(model_name):
     model_name = model_name.lower()
-    if 'rwkv-' in model_name.lower():
+    if 'rwkv-' in model_name:
         return 'rwkv'
     elif len(list(Path(f'{shared.args.model_dir}/{model_name}').glob('*ggml*.bin'))) > 0:
         return 'llamacpp'