fix for when sample_packing and eval_sample_packing are different (#1695)
Browse files
src/axolotl/utils/config/models/input/v0_4_1/__init__.py
CHANGED
@@ -908,6 +908,17 @@ class AxolotlInputConfig(
|
|
908 |
"explicitly setting `eval_sample_packing` to match `sample_packing`"
|
909 |
)
|
910 |
data["eval_sample_packing"] = True
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
911 |
return data
|
912 |
|
913 |
@model_validator(mode="before")
|
|
|
908 |
"explicitly setting `eval_sample_packing` to match `sample_packing`"
|
909 |
)
|
910 |
data["eval_sample_packing"] = True
|
911 |
+
|
912 |
+
if (
|
913 |
+
data.get("sample_packing")
|
914 |
+
and data.get("eval_sample_packing") is False
|
915 |
+
and data.get("remove_unused_columns") is None
|
916 |
+
):
|
917 |
+
LOG.info(
|
918 |
+
"setting `remove_unused_columns: false` for when sample_packing and eval_sample_packing don't match"
|
919 |
+
)
|
920 |
+
data["remove_unused_columns"] = False
|
921 |
+
|
922 |
return data
|
923 |
|
924 |
@model_validator(mode="before")
|