tocmo0nlord/axolotl
1
0
Fork 0
Code Issues Pull Requests Actions 2 Packages Projects Releases Wiki Activity

fix: parse model_kwargs (#1825)

Browse Source
This commit is contained in:
NanoCode012
2024-08-16 20:51:19 +09:00
committed by GitHub
parent f18925fb4b
commit 68a3c7678a
1 changed files with 2 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
src/axolotl/utils/config/models/input/v0_4_1/__init__.py
View File

@@ -321,6 +321,8 @@ class ModelInputConfig(BaseModel):
) )
trust_remote_code: Optional[bool] = None trust_remote_code: Optional[bool] = None
model_kwargs: Optional[Dict[str, Any]] = None
@field_validator("trust_remote_code") @field_validator("trust_remote_code")
@classmethod @classmethod
def hint_trust_remote_code(cls, trust_remote_code): def hint_trust_remote_code(cls, trust_remote_code):