fix: add missing config to schema

2025-05-13 14:24:33 +07:00
parent 7fa1089cea
commit 6db60ac520
1 changed files with 1 additions and 0 deletions
--- a/src/axolotl/utils/schemas/config.py
+++ b/src/axolotl/utils/schemas/config.py
@@ -83,6 +83,7 @@ class AxolotlInputConfig(
    # optionally shrink the embeddings when the tokenizer vocab size is smaller
    shrink_embeddings: bool | None = None
    embeddings_skip_upcast: bool | None = None
+    random_init_weights: bool | None = None

    rl: RLType | None = None
    trl: TRLConfig | None = Field(