From 14163c15d9d52e6fd8674eeb865f689735e42b67 Mon Sep 17 00:00:00 2001
From: Wing Lian <wing.lian@gmail.com>
Date: Sat, 10 Jun 2023 14:11:13 -0400
Subject: [PATCH] fix for local variable 'LlamaForCausalLM' referenced before
 assignment

---
 src/axolotl/utils/models.py | 1 +
 1 file changed, 1 insertion(+)

diff --git a/src/axolotl/utils/models.py b/src/axolotl/utils/models.py
index 8ceaa0d53..4a3b4c55a 100644
--- a/src/axolotl/utils/models.py
+++ b/src/axolotl/utils/models.py
@@ -90,6 +90,7 @@ def load_model(
     Load a model from a base model and a model type.
     """
 
+    global LlamaForCausalLM  # pylint: disable=global-statement
     # TODO refactor as a kwarg
     load_in_8bit = cfg.load_in_8bit
     cfg.is_llama_derived_model = "llama" in base_model or (