Spaces:

Dovakiins
/

qwerrwe

Build error

winglian commited on Feb 26, 2024

Commit

e7eed20

unverified ·

1 Parent(s): cf00231

hotfix for missing outputs params (#1333)

Files changed (1) hide show

src/axolotl/utils/config/models/input/v0_4_1/__init__.py CHANGED Viewed

@@ -336,6 +336,7 @@ class WandbConfig(BaseModel):
 # pylint: disable=too-many-public-methods,too-many-ancestors
 class AxolotlInputConfig(
     ModelInputConfig,
     LoraConfig,
     ReLoRAConfig,
     HyperparametersConfig,

 # pylint: disable=too-many-public-methods,too-many-ancestors
 class AxolotlInputConfig(
     ModelInputConfig,
+    ModelOutputConfig,
     LoraConfig,
     ReLoRAConfig,
     HyperparametersConfig,