Archives/text-generation-webui-docker
2
0
Fork 0
mirror of https://github.com/Atinoda/text-generation-webui-docker synced 2024-11-11 13:11:20 +00:00
Code Issues Packages Projects Releases Wiki Activity

Bump torch versions

Browse Source 
Closes #44
This commit is contained in:
Atinoda 2024-03-11 21:39:11 +00:00
parent 994e701bf2
commit 6726a71bf6
1 changed files with 4 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
Dockerfile
View File

@ -37,7 +37,7 @@ RUN cp -ar /src /app
# Base # Base
FROM app_base AS app_nvidia FROM app_base AS app_nvidia
# Install pytorch for CUDA 12.1 # Install pytorch for CUDA 12.1
RUN pip3 install torch==2.1.* torchvision==0.16.* torchaudio==2.1.* \ RUN pip3 install torch==2.2.1 torchvision==0.17.1 torchaudio==2.2.1 \
--index-url https://download.pytorch.org/whl/cu121 --index-url https://download.pytorch.org/whl/cu121
# Install oobabooga/text-generation-webui # Install oobabooga/text-generation-webui
RUN pip3 install -r /app/requirements.txt RUN pip3 install -r /app/requirements.txt
@ -53,7 +53,7 @@ RUN chmod +x /scripts/build_extensions.sh && \
# Base # Base
FROM app_base AS app_rocm FROM app_base AS app_rocm
# Install pytorch for ROCM # Install pytorch for ROCM
RUN pip3 install torch==2.1.* torchvision==0.16.* torchaudio==2.1.* \ RUN pip3 install torch==2.2.1 torchvision==0.17.1 torchaudio==2.2.1 \
--index-url https://download.pytorch.org/whl/rocm5.6 --index-url https://download.pytorch.org/whl/rocm5.6
# Install oobabooga/text-generation-webui # Install oobabooga/text-generation-webui
RUN pip3 install -r /app/requirements_amd.txt RUN pip3 install -r /app/requirements_amd.txt
@ -89,7 +89,7 @@ RUN chmod +x /scripts/build_extensions.sh && \
# Base # Base
FROM app_base AS app_cpu FROM app_base AS app_cpu
# Install pytorch for CPU # Install pytorch for CPU
RUN pip3 install torch==2.1.* torchvision==0.16.* torchaudio==2.1.* \ RUN pip3 install torch==2.2.1 torchvision==0.17.1 torchaudio==2.2.1 \
--index-url https://download.pytorch.org/whl/cpu --index-url https://download.pytorch.org/whl/cpu
# Install oobabooga/text-generation-webui # Install oobabooga/text-generation-webui
RUN pip3 install -r /app/requirements_cpu_only.txt RUN pip3 install -r /app/requirements_cpu_only.txt
@ -217,4 +217,4 @@ COPY --from=app_cpu_x $VIRTUAL_ENV $VIRTUAL_ENV
# Variant parameters # Variant parameters
RUN echo "CPU Extended" > /variant.txt RUN echo "CPU Extended" > /variant.txt
ENV EXTRA_LAUNCH_ARGS="" ENV EXTRA_LAUNCH_ARGS=""
CMD ["python3", "/app/server.py"] CMD ["python3", "/app/server.py"]