Skip to content

Commit

Permalink
Bumped to latest oobabooga commit, Added screen and tmux, reverted EN…
Browse files Browse the repository at this point in the history
…TRYPOINT to CMD so that docker command can be overridden on RunPod.
  • Loading branch information
ashleykleynhans committed Feb 5, 2024
1 parent 495142d commit a1c3e20
Show file tree
Hide file tree
Showing 2 changed files with 13 additions and 6 deletions.
14 changes: 9 additions & 5 deletions Dockerfile
Original file line number Diff line number Diff line change
@@ -1,7 +1,7 @@
# Stage 1: Base
FROM nvidia/cuda:12.1.1-cudnn8-devel-ubuntu22.04 as base

ARG OOBABOOGA_COMMIT=837bd888e4cf239094d9b1cabcc342266fee11c0
ARG OOBABOOGA_COMMIT=4f3fdf1b5ff6884b9899a3630b3ed9aae27decbf
ARG TORCH_VERSION=2.1.2

SHELL ["/bin/bash", "-o", "pipefail", "-c"]
Expand Down Expand Up @@ -41,10 +41,14 @@ RUN apt update && \
zip \
unzip \
htop \
screen \
tmux \
pkg-config \
libcairo2-dev \
libgoogle-perftools4 libtcmalloc-minimal4 \
apt-transport-https ca-certificates && \
libgoogle-perftools4 \
libtcmalloc-minimal4 \
apt-transport-https \
ca-certificates && \
update-ca-certificates && \
apt clean && \
rm -rf /var/lib/apt/lists/* && \
Expand Down Expand Up @@ -126,6 +130,6 @@ WORKDIR /
COPY --chmod=755 scripts/* ./

# Start the container
ENV TEMPLATE_VERSION=1.11.0
ENV TEMPLATE_VERSION=1.11.1
SHELL ["/bin/bash", "--login", "-c"]
ENTRYPOINT [ "/start.sh" ]
CMD [ "/start.sh" ]
5 changes: 4 additions & 1 deletion nginx/template-readme.md
Original file line number Diff line number Diff line change
@@ -1,6 +1,6 @@
## Text Generation Web UI: A Gradio web UI for Large Language Models. Supports transformers, GPTQ, llama.cpp (GGUF), Llama models

### Version 1.11.0
### Version 1.11.1

### Included in this Template

Expand All @@ -15,6 +15,9 @@
* [runpodctl](https://github.com/runpod/runpodctl)
* [croc](https://github.com/schollz/croc)
* [rclone](https://rclone.org/)
* speedtest-cli
* screen
* tmux

### Ports

Expand Down

0 comments on commit a1c3e20

Please sign in to comment.