Zac Brannelly commited on
Commit
73f1bda
1 Parent(s): 1c7ed26

Fix bug preventing model_kwargs being injected (#1262)

Browse files
Files changed (1) hide show
  1. src/axolotl/utils/models.py +1 -1
src/axolotl/utils/models.py CHANGED
@@ -400,7 +400,7 @@ def load_model(
400
  model_kwargs: Dict[str, Any] = {}
401
 
402
  if cfg.model_kwargs:
403
- for key, val in model_kwargs.items():
404
  model_kwargs[key] = val
405
 
406
  max_memory = cfg.max_memory
 
400
  model_kwargs: Dict[str, Any] = {}
401
 
402
  if cfg.model_kwargs:
403
+ for key, val in cfg.model_kwargs.items():
404
  model_kwargs[key] = val
405
 
406
  max_memory = cfg.max_memory