fix: parse model_kwargs (#1825)

axolotl-ai-cloud · Aug 16, 2024 · 68a3c76 · 68a3c76
1 parent f18925f
commit 68a3c76
Showing 1 changed file with 2 additions and 0 deletions.
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
@@ -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):