fix

Lightning-AI · Mar 8, 2024 · 8782a4d · 8782a4d
1 parent 2e918d5
commit 8782a4d
Showing 1 changed file with 1 addition and 0 deletions.
diff --git 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