[fine_tuning] testing: config: give a GPU to the light profile

openshift-psap · Nov 24, 2024 · 50f14a3 · 50f14a3
1 parent f5d826e
commit 50f14a3
Showing 1 changed file with 3 additions and 1 deletion.
diff --git 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