From 483defb9ae7777a7c1a3971982776ade6547757a Mon Sep 17 00:00:00 2001 From: Wing Lian Date: Tue, 14 Jan 2025 23:56:00 -0500 Subject: [PATCH] try tests for kd on l40s --- cicd/tests.py | 2 +- tests/e2e/integrations/test_kd.py | 1 + 2 files changed, 2 insertions(+), 1 deletion(-) diff --git a/cicd/tests.py b/cicd/tests.py index d7ae5b5e8..b8761ba53 100644 --- a/cicd/tests.py +++ b/cicd/tests.py @@ -59,7 +59,7 @@ VOLUME_CONFIG = { } N_GPUS = int(os.environ.get("N_GPUS", 1)) -GPU_CONFIG = modal.gpu.A10G(count=N_GPUS) +GPU_CONFIG = modal.gpu.L40S(count=N_GPUS) def run_cmd(cmd: str, run_folder: str): diff --git a/tests/e2e/integrations/test_kd.py b/tests/e2e/integrations/test_kd.py index 18987b68b..7eb08afe3 100644 --- a/tests/e2e/integrations/test_kd.py +++ b/tests/e2e/integrations/test_kd.py @@ -33,6 +33,7 @@ def min_cfg(temp_dir): "dataloader_prefetch_factor": 8, "dataloader_num_workers": 4, "dataloader_pin_memory": True, + # "dataset_prepared_path": str(Path(temp_dir) / "last_run_prepared"), "datasets": [ { "path": "axolotl-ai-co/evolkit-logprobs-pipeline-75k-v2-sample",