From e285e24f7f4e1718c0f26c93eb7eec8e669ae97e Mon Sep 17 00:00:00 2001
From: NanoCode012 <kevinvong@rocketmail.com>
Date: Sun, 11 Jun 2023 10:52:12 +0900
Subject: [PATCH] Address PR suggestion

Co-authored-by: Wing Lian <wing.lian@gmail.com>
---
 src/axolotl/utils/models.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/src/axolotl/utils/models.py b/src/axolotl/utils/models.py
index b84597076..b3a5eeb60 100644
--- a/src/axolotl/utils/models.py
+++ b/src/axolotl/utils/models.py
@@ -202,7 +202,7 @@ def load_model(
                 else True,
             )
             load_in_8bit = False
-        elif cfg.is_llama_derived_model and "LlamaForCausalLM" in globals():
+        elif cfg.is_llama_derived_model:
             try:
                 from transformers import LlamaForCausalLM
             except ImportError: