tocmo0nlord/axolotl
1
0
Fork 0
Code Issues Pull Requests Actions 3 Packages Projects Releases Wiki Activity

Address PR suggestion

Browse Source 
Co-authored-by: Wing Lian <wing.lian@gmail.com>
This commit is contained in:
NanoCode012
2023-06-11 10:52:12 +09:00
committed by GitHub
parent 919727b4d7
commit e285e24f7f
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
src/axolotl/utils/models.py
View File

@@ -202,7 +202,7 @@ def load_model(
else True, else True,
) )
load_in_8bit = False load_in_8bit = False
elif cfg.is_llama_derived_model and "LlamaForCausalLM" in globals(): elif cfg.is_llama_derived_model:
try: try:
from transformers import LlamaForCausalLM from transformers import LlamaForCausalLM
except ImportError: except ImportError: