diff --git a/src/axolotl/utils/models.py b/src/axolotl/utils/models.py index 44a93a36b..e1748fd7b 100644 --- a/src/axolotl/utils/models.py +++ b/src/axolotl/utils/models.py @@ -400,7 +400,7 @@ def load_model( model_kwargs: Dict[str, Any] = {} if cfg.model_kwargs: - for key, val in model_kwargs.items(): + for key, val in cfg.model_kwargs.items(): model_kwargs[key] = val max_memory = cfg.max_memory