From 6db60ac5202673f4dd484b3fa9458c3302eb5d6b Mon Sep 17 00:00:00 2001 From: NanoCode012 Date: Tue, 13 May 2025 14:24:33 +0700 Subject: [PATCH] fix: add missing config to schema --- src/axolotl/utils/schemas/config.py | 1 + 1 file changed, 1 insertion(+) diff --git a/src/axolotl/utils/schemas/config.py b/src/axolotl/utils/schemas/config.py index 25c802959..e64235956 100644 --- 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(