Zac Brannelly
commited on
Fix bug preventing model_kwargs being injected (#1262)
Browse files
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
|