winglian commited on
Commit
18cabc0
·
unverified ·
1 Parent(s): ed8ef65

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")