diff --git a/Dockerfile b/Dockerfile
index 51e3401..8a29a23 100644
--- a/Dockerfile
+++ b/Dockerfile
@@ -9,4 +9,4 @@ RUN pip3 install -r requirements.txt
 WORKDIR /app/src
 
 
-CMD ["sh", "-c", "bash start.sh --hf_token ${HF_TOKEN} --flock_api_key ${FLOCK_API_KEY} --task_id ${TASK_ID}"]
\ No newline at end of file
+CMD ["sh", "-c", "bash start.sh --hf_token ${HF_TOKEN} --flock_api_key ${FLOCK_API_KEY} --task_id ${TASK_ID} --validation_args_file validation_config_cpu.json.example"]
\ No newline at end of file
diff --git a/Dockerfile-gpu b/Dockerfile-gpu
index 53a8760..62e35f5 100644
--- a/Dockerfile-gpu
+++ b/Dockerfile-gpu
@@ -17,4 +17,4 @@ WORKDIR /app/src
 
 ENV IS_DOCKER_CONTAINER=1
 
-CMD ["sh", "-c", "bash start.sh --hf_token ${HF_TOKEN} --flock_api_key ${FLOCK_API_KEY} --task_id ${TASK_ID}"]
\ No newline at end of file
+CMD ["sh", "-c", "bash start.sh --hf_token ${HF_TOKEN} --flock_api_key ${FLOCK_API_KEY} --task_id ${TASK_ID} --validation_args_file validation_config.json.example"]
\ No newline at end of file