Add missing tei-dlc after removing defaults

Author: alvarobartt

.github/workflows/run-tests-action.yml

@@ -19,6 +19,10 @@ on:
         description: "The URI of the Hugging Face TGI DLC (GPU only)."
         required: false
         type: string
+      tei-dlc:
+        description: "The URI of the Hugging Face TEI DLC (CPU and GPU)."
+        required: true
+        type: string
 
 jobs:
   run-tests:
@@ -51,3 +55,4 @@ jobs:
           TRAINING_DLC: ${{ inputs.training-dlc }}
           INFERENCE_DLC: ${{ inputs.inference-dlc }}
           TGI_DLC: ${{ inputs.tgi-dlc }}
+          TEI_DLC: ${{ inputs.tei-dlc }}

.github/workflows/test-huggingface-dlcs.yml

@@ -28,6 +28,7 @@ jobs:
     with:
       group: aws-general-8-plus
       inference-dlc: us-docker.pkg.dev/deeplearning-platform-release/gcr.io/huggingface-pytorch-inference-cpu.2-2.transformers.4-44.ubuntu2204.py311
+      tei-dlc: us-docker.pkg.dev/deeplearning-platform-release/gcr.io/huggingface-text-embeddings-inference-cpu.1-2
 
   dlcs-on-gpu:
     name: Run Hugging Face DLCs Tests on GPU
@@ -37,3 +38,4 @@ jobs:
       training-dlc: us-docker.pkg.dev/deeplearning-platform-release/gcr.io/huggingface-pytorch-training-cu121.transformers.4-42.ubuntu2204.py310
       inference-dlc: us-docker.pkg.dev/deeplearning-platform-release/gcr.io/huggingface-pytorch-inference-cu121.2-2.transformers.4-44.ubuntu2204.py311
       tgi-dlc: us-docker.pkg.dev/deeplearning-platform-release/gcr.io/huggingface-text-generation-inference-cu121.2-2.ubuntu2204.py310
+      tei-dlc: us-docker.pkg.dev/deeplearning-platform-release/gcr.io/huggingface-text-embeddings-inference-cu122.1-4.ubuntu2204