tocmo0nlord/axolotl
1
0
Fork 0
Code Issues Pull Requests Actions 3 Packages Projects Releases Wiki Activity

fix

Browse Source
This commit is contained in:
Dan Saunders
2025-01-13 17:25:12 +00:00
parent 3b82fc36ec
commit a030dad657
2 changed files with 4 additions and 7 deletions
Download Patch File Download Diff File Expand all files Collapse all files

7
tests/e2e/test_mamba.py
View File

@@ -63,10 +63,5 @@ class TestMamba(unittest.TestCase):
cli_args = TrainerCliArgs()
dataset_meta = load_datasets(cfg=cfg, cli_args=cli_args)
<<<<<<< HEAD
train(cfg=cfg, cli_args=cli_args, dataset_meta=dataset_meta)
check_model_output_exists(temp_dir, cfg)
=======
train(cfg=cfg, dataset_meta=dataset_meta)
assert (Path(temp_dir) / "pytorch_model.bin").exists()
>>>>>>> 2a421127 (continued cleanup and documentation)
check_model_output_exists(temp_dir, cfg)