diff --git a/.github/workflows/train-model.yml b/.github/workflows/train-model.yml
index 2a08b0f6..3ae0c428 100644
--- a/.github/workflows/train-model.yml
+++ b/.github/workflows/train-model.yml
@@ -166,6 +166,8 @@ jobs:
               persistentVolumeClaim:
                 claimName: task-pvc
             params:
+            - name: ENERGY_SOURCE
+              value: ${{ inputs.energy_source }}
             - name: MODEL_SERVER_IMAGE
               value: ${{ inputs.model_server_image }}
             - name: PIPELINE_NAME