Fix: eval table conflict with eval_sample_packing (#769)
Browse files- src/axolotl/utils/config.py +9 -0
- tests/test_validation.py +41 -0
src/axolotl/utils/config.py
CHANGED
@@ -348,6 +348,15 @@ def validate_config(cfg):
|
|
348 |
"eval_steps and evaluation_strategy are not supported with val_set_size == 0"
|
349 |
)
|
350 |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
351 |
# TODO
|
352 |
# MPT 7b
|
353 |
# https://github.com/facebookresearch/bitsandbytes/issues/25
|
|
|
348 |
"eval_steps and evaluation_strategy are not supported with val_set_size == 0"
|
349 |
)
|
350 |
|
351 |
+
if (
|
352 |
+
cfg.sample_packing
|
353 |
+
and cfg.eval_table_size
|
354 |
+
and cfg.eval_sample_packing is not False
|
355 |
+
):
|
356 |
+
raise ValueError(
|
357 |
+
"eval_table_size and eval_sample_packing are not supported together with sample_packing. Please set 'eval_sample_packing' to false."
|
358 |
+
)
|
359 |
+
|
360 |
# TODO
|
361 |
# MPT 7b
|
362 |
# https://github.com/facebookresearch/bitsandbytes/issues/25
|
tests/test_validation.py
CHANGED
@@ -565,3 +565,44 @@ class ValidationTest(unittest.TestCase):
|
|
565 |
)
|
566 |
|
567 |
validate_config(cfg)
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
565 |
)
|
566 |
|
567 |
validate_config(cfg)
|
568 |
+
|
569 |
+
def test_eval_table_size_conflict_eval_packing(self):
|
570 |
+
cfg = DictDefault(
|
571 |
+
{
|
572 |
+
"sample_packing": True,
|
573 |
+
"eval_table_size": 100,
|
574 |
+
}
|
575 |
+
)
|
576 |
+
|
577 |
+
with pytest.raises(
|
578 |
+
ValueError, match=r".*Please set 'eval_sample_packing' to false.*"
|
579 |
+
):
|
580 |
+
validate_config(cfg)
|
581 |
+
|
582 |
+
cfg = DictDefault(
|
583 |
+
{
|
584 |
+
"sample_packing": True,
|
585 |
+
"eval_sample_packing": False,
|
586 |
+
}
|
587 |
+
)
|
588 |
+
|
589 |
+
validate_config(cfg)
|
590 |
+
|
591 |
+
cfg = DictDefault(
|
592 |
+
{
|
593 |
+
"sample_packing": False,
|
594 |
+
"eval_table_size": 100,
|
595 |
+
}
|
596 |
+
)
|
597 |
+
|
598 |
+
validate_config(cfg)
|
599 |
+
|
600 |
+
cfg = DictDefault(
|
601 |
+
{
|
602 |
+
"sample_packing": True,
|
603 |
+
"eval_table_size": 100,
|
604 |
+
"eval_sample_packing": False,
|
605 |
+
}
|
606 |
+
)
|
607 |
+
|
608 |
+
validate_config(cfg)
|