Remove fschat from Dockerfile

Fastchat is removed from all dependencies

Signed-off-by: kingbri <bdashore3@proton.me>
This commit is contained in:
kingbri 2023-12-17 21:45:23 -05:00 committed by Brian Dashore
parent de9a19b5d3
commit 124e39df26

View file

@ -7,12 +7,6 @@ RUN apt-get update && apt-get install -y \
python3-pip \
&& rm -rf /var/lib/apt/lists/*
# Define a build-time argument for conditional installation
ARG INSTALL_FSCHAT=false
# Set the environment variable based on the build argument
ENV INSTALL_FSCHAT=$INSTALL_FSCHAT
# Set the working directory in the container
WORKDIR /usr/src/app
@ -26,9 +20,6 @@ RUN pip install exllamav2
# Install any other needed packages specified in requirements.txt
RUN pip install --no-cache-dir -r requirements.txt
# Conditional installation of fschat[model_worker]
RUN if [ "$INSTALL_FSCHAT" = "true" ] ; then pip install fschat[model_worker] ; fi
# Copy the sample config file to the main config
RUN cp config_sample.yml config.yml