Nanobit winglian commited on
Commit
e285e24
1 Parent(s): 919727b

Address PR suggestion

Browse files

Co-authored-by: Wing Lian <wing.lian@gmail.com>

Files changed (1) hide show
  1. src/axolotl/utils/models.py +1 -1
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 and "LlamaForCausalLM" in globals():
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: