fix num_labels error and update the version of training operator cont…

…roller Signed-off-by: helenxie-bit <[email protected]>
kubeflow · Sep 3, 2024 · c860238 · c860238
1 parent 8461a49
commit c860238
Show file tree

Hide file tree

Showing 3 changed files with 4 additions and 2 deletions.
diff --git a/.github/workflows/e2e-test-tune-api.yaml b/.github/workflows/e2e-test-tune-api.yaml
@@ -24,7 +24,7 @@ jobs:
 
       - name: Install Training Operator SDK
         shell: bash
-        run: pip install kubeflow-training[huggingface]==1.8.0
+        run: pip install -U kubeflow-training[huggingface]
 
       - name: Run e2e test with tune API
         uses: ./.github/workflows/template-e2e-test

diff --git a/sdk/python/v1beta1/kubeflow/katib/api/katib_client.py b/sdk/python/v1beta1/kubeflow/katib/api/katib_client.py
@@ -633,6 +633,8 @@ class name in this argument.
                     model_provider_parameters.model_uri,
                     "--transformer_type",
                     model_provider_parameters.transformer_type.__name__,
+                    "--num_labels",
+                    str(model_provider_parameters.num_labels),
                     "--model_dir",
                     VOLUME_PATH_MODEL,
                     "--dataset_dir",

diff --git a/test/e2e/v1beta1/scripts/gh-actions/setup-katib.sh b/test/e2e/v1beta1/scripts/gh-actions/setup-katib.sh
@@ -25,7 +25,7 @@ DEPLOY_TRAINING_OPERATOR=${2:-false}
 WITH_DATABASE_TYPE=${3:-mysql}
 
 E2E_TEST_IMAGE_TAG="e2e-test"
-TRAINING_OPERATOR_VERSION="v1.6.0-rc.0"
+TRAINING_OPERATOR_VERSION="v1.8.0"
 
 echo "Start to install Katib"