From 4e188eeb80a1ca537523d46958f3f1deef48b2d2 Mon Sep 17 00:00:00 2001
From: oobabooga <112222186+oobabooga@users.noreply.github.com>
Date: Sat, 3 Feb 2024 20:39:50 -0800
Subject: [PATCH] Lint

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

diff --git a/modules/LoRA.py b/modules/LoRA.py
index 6be473e0..0cb1671e 100644
--- a/modules/LoRA.py
+++ b/modules/LoRA.py
@@ -12,7 +12,7 @@ from modules.models import reload_model
 def add_lora_to_model(lora_names):
     if 'GPTQForCausalLM' in shared.model.__class__.__name__ or shared.args.loader == 'AutoGPTQ':
         add_lora_autogptq(lora_names)
-    elif shared.model.__class__.__name__ in ['Exllamav2HF'] or shared.args.loader == ['ExLlamav2_HF']:
+    elif shared.model.__class__.__name__ == 'Exllamav2HF' or shared.args.loader == 'ExLlamav2_HF':
         add_lora_exllamav2(lora_names)
     else:
         add_lora_transformers(lora_names)