From 68a3c7678a3cbf5098239f84a548c9492ebe4387 Mon Sep 17 00:00:00 2001 From: NanoCode012 Date: Fri, 16 Aug 2024 20:51:19 +0900 Subject: [PATCH] fix: parse model_kwargs (#1825) --- src/axolotl/utils/config/models/input/v0_4_1/__init__.py | 2 ++ 1 file changed, 2 insertions(+) diff --git a/src/axolotl/utils/config/models/input/v0_4_1/__init__.py b/src/axolotl/utils/config/models/input/v0_4_1/__init__.py index 7bf77965c..647d6b88c 100644 --- a/src/axolotl/utils/config/models/input/v0_4_1/__init__.py +++ b/src/axolotl/utils/config/models/input/v0_4_1/__init__.py @@ -321,6 +321,8 @@ class ModelInputConfig(BaseModel): ) trust_remote_code: Optional[bool] = None + model_kwargs: Optional[Dict[str, Any]] = None + @field_validator("trust_remote_code") @classmethod def hint_trust_remote_code(cls, trust_remote_code):