hotfix for missing outputs params (#1333)

2024-02-26 14:36:37 -05:00
parent cf002312e0
commit e7eed203d8
1 changed files with 1 additions and 0 deletions
--- a/src/axolotl/utils/config/models/input/v0_4_1/init.py
+++ b/src/axolotl/utils/config/models/input/v0_4_1/init.py
@@ -336,6 +336,7 @@ class WandbConfig(BaseModel):
 # pylint: disable=too-many-public-methods,too-many-ancestors
 class AxolotlInputConfig(
    ModelInputConfig,
+    ModelOutputConfig,
    LoraConfig,
    ReLoRAConfig,
    HyperparametersConfig,