Update Dockerfile

This commit is contained in:
simonmcnair 2025-05-05 09:58:02 +01:00 committed by GitHub
parent 087b5bffa1
commit b959dcf21b
No known key found for this signature in database
GPG key ID: B5690EEEBB952194

View file

@ -60,4 +60,6 @@ RUN chmod u+x /docker/entrypoint.sh && cp /docker/extra_model_paths.yaml ${ROOT}
ENV NVIDIA_VISIBLE_DEVICES=all PYTHONPATH="${PYTHONPATH}:${PWD}" CLI_ARGS=""
EXPOSE 7860
ENTRYPOINT ["/docker/entrypoint.sh"]
CMD python -u main.py --listen --port 7860 ${CLI_ARGS}
#CMD python -u main.py --listen --port 7860 ${CLI_ARGS}
CMD sh -c "python3 -c 'import os; print(f\"Running as UID={os.getuid()}, GID={os.getgid()}\")' && python -u main.py --listen --port 7860 ${CLI_ARGS}"