Skip to content

Commit

Permalink
bump upstream version
Browse files Browse the repository at this point in the history
  • Loading branch information
mlipscombe committed Sep 29, 2024
1 parent 6c8af1c commit 5352354
Showing 1 changed file with 2 additions and 2 deletions.
4 changes: 2 additions & 2 deletions Dockerfile
Original file line number Diff line number Diff line change
@@ -1,5 +1,5 @@
FROM pytorch/pytorch:2.4.0-cuda12.4-cudnn9-runtime
ARG TAG_VERSION=v0.0.6
ARG TAG_VERSION=v0.2.2

ENV DEBIAN_FRONTEND=noninteractive PIP_PREFER_BINARY=1

Expand All @@ -19,4 +19,4 @@ 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}

0 comments on commit 5352354

Please sign in to comment.