Zac Brannelly
commited on
Commit
•
73f1bda
1
Parent(s):
1c7ed26
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
|