From 50f14a386ff2006f9a4629fc67c23b2a945a8307 Mon Sep 17 00:00:00 2001 From: Kevin Pouget Date: Sat, 23 Nov 2024 10:01:46 +0100 Subject: [PATCH] [fine_tuning] testing: config: give a GPU to the light profile --- projects/fine_tuning/testing/config.yaml | 4 +++- 1 file changed, 3 insertions(+), 1 deletion(-) diff --git a/projects/fine_tuning/testing/config.yaml b/projects/fine_tuning/testing/config.yaml index f0862ac84..caae24dc4 100644 --- a/projects/fine_tuning/testing/config.yaml +++ b/projects/fine_tuning/testing/config.yaml @@ -344,9 +344,11 @@ ci_presets: # --- ilab: - extends: [not_fms] + extends: [not_fms, gpu] ci_presets.light["tests.fine_tuning.test_settings.dataset_name"]: ilab_skills_data.jsonl ci_presets.light["tests.fine_tuning.test_settings.model_name"]: ibm-granite/granite-3b-code-instruct@hf + ci_presets.light["tests.fine_tuning.test_settings.gpu"]: 1 + ci_presets.gpu["clusters.sutest.compute.machineset.type"]: g5.xlarge tests.fine_tuning.ilab.enabled: true tests.fine_tuning.test_settings.name: ilab tests.fine_tuning.test_settings.dataset_name: ilab_skills_data.jsonl