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

Merge pull request #182 from OpenAccess-AI-Collective/fix-llama-ref

Browse Source 
fix for local variable 'LlamaForCausalLM' referenced before assignment
This commit is contained in:
Wing Lian
2023-06-10 14:25:51 -04:00
committed by GitHub
parent 41e4f6ca31 14163c15d9
commit 01248253a3
1 changed files with 1 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

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

@@ -90,6 +90,7 @@ def load_model(
Load a model from a base model and a model type. Load a model from a base model and a model type.
""" """
global LlamaForCausalLM # pylint: disable=global-statement
# TODO refactor as a kwarg # TODO refactor as a kwarg
load_in_8bit = cfg.load_in_8bit load_in_8bit = cfg.load_in_8bit
cfg.is_llama_derived_model = "llama" in base_model or ( cfg.is_llama_derived_model = "llama" in base_model or (