Address PR suggestion
Browse filesCo-authored-by: Wing Lian <wing.lian@gmail.com>
src/axolotl/utils/models.py
CHANGED
@@ -202,7 +202,7 @@ def load_model(
|
|
202 |
else True,
|
203 |
)
|
204 |
load_in_8bit = False
|
205 |
-
elif cfg.is_llama_derived_model
|
206 |
try:
|
207 |
from transformers import LlamaForCausalLM
|
208 |
except ImportError:
|
|
|
202 |
else True,
|
203 |
)
|
204 |
load_in_8bit = False
|
205 |
+
elif cfg.is_llama_derived_model:
|
206 |
try:
|
207 |
from transformers import LlamaForCausalLM
|
208 |
except ImportError:
|