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

fix: add missing config to schema

Browse Source
This commit is contained in:
NanoCode012
2025-05-13 14:24:33 +07:00
parent 7fa1089cea
commit 6db60ac520
1 changed files with 1 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

1
src/axolotl/utils/schemas/config.py
View File

@@ -83,6 +83,7 @@ class AxolotlInputConfig(
# optionally shrink the embeddings when the tokenizer vocab size is smaller # optionally shrink the embeddings when the tokenizer vocab size is smaller
shrink_embeddings: bool | None = None shrink_embeddings: bool | None = None
embeddings_skip_upcast: bool | None = None embeddings_skip_upcast: bool | None = None
random_init_weights: bool | None = None
rl: RLType | None = None rl: RLType | None = None
trl: TRLConfig | None = Field( trl: TRLConfig | None = Field(