Update Dockerfile

runpod-workers · May 10, 2023 · 9f24389 · 9f24389
1 parent 1c27df4
commit 9f24389
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/Dockerfile b/Dockerfile
@@ -26,4 +26,4 @@ RUN rm /model_fetcher.py
 # Add src files (Worker Template)
 ADD src .
 
-CMD [ "python", "-u", "/rp_handler.py", "--model_tag=${MODEL_TAG}" ]
+CMD python -u /rp_handler.py --model_tag=${MODEL_TAG}