fix for local variable 'LlamaForCausalLM' referenced before assignment
Browse files
src/axolotl/utils/models.py
CHANGED
|
@@ -90,6 +90,7 @@ def load_model(
|
|
| 90 |
Load a model from a base model and a model type.
|
| 91 |
"""
|
| 92 |
|
|
|
|
| 93 |
# TODO refactor as a kwarg
|
| 94 |
load_in_8bit = cfg.load_in_8bit
|
| 95 |
cfg.is_llama_derived_model = "llama" in base_model or (
|
|
|
|
| 90 |
Load a model from a base model and a model type.
|
| 91 |
"""
|
| 92 |
|
| 93 |
+
global LlamaForCausalLM # pylint: disable=global-statement
|
| 94 |
# TODO refactor as a kwarg
|
| 95 |
load_in_8bit = cfg.load_in_8bit
|
| 96 |
cfg.is_llama_derived_model = "llama" in base_model or (
|