From 8782a4d2aac0ab9e56f55d2361cc0911e9dfc001 Mon Sep 17 00:00:00 2001 From: awaelchli Date: Fri, 8 Mar 2024 21:03:28 +0100 Subject: [PATCH] fix --- tests/test_merge_lora.py | 1 + 1 file changed, 1 insertion(+) diff --git a/tests/test_merge_lora.py b/tests/test_merge_lora.py index 5cab0c37e3..6855b55599 100644 --- a/tests/test_merge_lora.py +++ b/tests/test_merge_lora.py @@ -25,6 +25,7 @@ def test_merge_lora(tmp_path, fake_checkpoint_dir): lora_checkpoint_dir = tmp_path / "lora" shutil.copytree(fake_checkpoint_dir, pretrained_checkpoint_dir) shutil.copytree(fake_checkpoint_dir, lora_checkpoint_dir) + (lora_checkpoint_dir / "lit_model.pth").unlink() # should not already exist shutil.rmtree(tmp_path / "checkpoints") # Create a fake pretrained checkpoint