Add QLoRA sanity test for training operator

red-hat-data-services · Sep 24, 2024 · 0fa961a · 0fa961a
1 parent ce4d40f
commit 0fa961a
Show file tree

Hide file tree

Showing 2 changed files with 11 additions and 1 deletion.
diff --git a/ods_ci/tests/Resources/Page/DistributedWorkloads/DistributedWorkloads.resource b/ods_ci/tests/Resources/Page/DistributedWorkloads/DistributedWorkloads.resource
@@ -10,7 +10,7 @@ ${CODEFLARE-SDK-API_URL}                 %{CODEFLARE-SDK-API_URL=https://api.git
 ${CODEFLARE-SDK_DIR}                     codeflare-sdk
 ${CODEFLARE-SDK_REPO_URL}                %{CODEFLARE-SDK_REPO_URL=https://github.com/project-codeflare/codeflare-sdk.git}
 ${DISTRIBUTED_WORKLOADS_RELEASE_ASSETS}  https://github.com/opendatahub-io/distributed-workloads/releases/latest/download
-${FMS_HF_TUNING_IMAGE}                   quay.io/modh/fms-hf-tuning@sha256:8edea6f0f9c4c631cdca1e1c10abf0d4b994738fde78c40d48eda216fdd382f5
+${FMS_HF_TUNING_IMAGE}                   quay.io/modh/fms-hf-tuning@sha256:f8f732c340488734bf9023953d14bb2410991bd3ff2a519ad2ce07c531353797
 ${NOTEBOOK_IMAGE}                        quay.io/modh/odh-generic-data-science-notebook@sha256:9d7f80080a453bcf7dee01b986df9ee811ee74f6f433c601a8b67d283c160547
 ${NOTEBOOK_USER_NAME}                    ${TEST_USER_3.USERNAME}
 ${NOTEBOOK_USER_PASSWORD}                ${TEST_USER_3.PASSWORD}

diff --git a/.../tests/Tests/0600__distributed_workloads/0602__training/test-run-tuning-stack-tests.robot b/.../tests/Tests/0600__distributed_workloads/0602__training/test-run-tuning-stack-tests.robot
@@ -27,6 +27,16 @@ Run Training operator ODH test base LoRA use case
     ...     TrainingOperator
     Run Training Operator ODH Core Test    TestPytorchjobWithSFTtrainerLoRa
 
+Run Training operator ODH test base QLoRA use case
+    [Documentation]    Run Go ODH tests for Training operator base QLoRA use case
+    [Tags]  RHOAIENG-13142
+    ...     Resources-GPU
+    ...     Tier1
+    ...     DistributedWorkloads
+    ...     Training
+    ...     TrainingOperator
+    Run Training Operator ODH Core Test    TestPytorchjobWithSFTtrainerQLoRa
+
 Run Training operator ODH test with Kueue quota
     [Documentation]    Run Go ODH tests for Training operator with Kueue quota
     [Tags]  RHOAIENG-6965