fix: entrypoint for Dockerfile (#33)

FLock-io · Jun 20, 2024 · 53abc51 · 53abc51
1 parent fa1a4da
commit 53abc51
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git 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}"]
+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"]
diff --git 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}"]
+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"]
Original file line number	Diff line number	Diff line change
Expand Up		@@ -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}"]
		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"]
Original file line number	Diff line number	Diff line change
Expand Up		@@ -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}"]
		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"]