From 42d3e36a6f5d980d4de850bc85c398573afea737 Mon Sep 17 00:00:00 2001 From: Wing Lian Date: Wed, 8 Jan 2025 20:15:42 -0500 Subject: [PATCH] fix adapter model check --- tests/e2e/integrations/test_kd.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/tests/e2e/integrations/test_kd.py b/tests/e2e/integrations/test_kd.py index a7e9ee494..de6b1dcac 100644 --- a/tests/e2e/integrations/test_kd.py +++ b/tests/e2e/integrations/test_kd.py @@ -114,7 +114,7 @@ class TestKnowledgeDistillation: dataset_meta = load_datasets(cfg=cfg, cli_args=cli_args) train(cfg=cfg, cli_args=cli_args, dataset_meta=dataset_meta) - assert (Path(temp_dir) / "model.safetensors").exists() + assert (Path(temp_dir) / "adapter_model.safetensors").exists() check_tensorboard( temp_dir + "/runs", "train/loss", 1.0, "Train Loss is too high" )