fix: parse model_kwargs (#1825)

2024-08-16 20:51:19 +09:00
parent f18925fb4b
commit 68a3c7678a
1 changed files with 2 additions and 0 deletions
--- 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):