Merge pull request #1546 from PAIR-code:lit-on-gcp

PiperOrigin-RevId: 686309136

Author: LIT team

.github/workflows/ci.yml

@@ -45,14 +45,10 @@ jobs:
       uses: actions/setup-python@v4
       with:
         python-version: ${{ matrix.python-version }}
-    - name: Install Python dependencies
-      run: python -m pip install -r requirements.txt
-    - name: Install LIT package
-      run: python -m pip install -e .
+    - name: Install LIT package with testing dependencies
+      run: python -m pip install -e '.[test]'
     - name: Test LIT
-      run: |
-        python -m pip install pytest
-        pytest -v
+      run: pytest -v
     - name: Setup Node ${{ matrix.node-version }}
       uses: actions/setup-node@v2
       with:
@@ -73,4 +69,5 @@ jobs:
     - name: Build Docker image
       uses: docker/build-push-action@v4
       with:
+        target: lit-nlp-prod
         tags: lit-nlp:ci-${{ github.sha }}

Dockerfile

@@ -14,26 +14,51 @@
 # ==============================================================================
 # Use the official lightweight Python image.
 # https://hub.docker.com/_/python
-FROM python:3.10-slim
+
+# ---- LIT Base Container ----
+
+FROM python:3.11-slim AS lit-nlp-base
 
 # Update Ubuntu packages and install basic utils
 RUN apt-get update
 RUN apt-get install -y wget curl gnupg2 gcc g++ git
 
+# Copy local code to the container image.
+ENV APP_HOME=/app
+WORKDIR $APP_HOME
+
+COPY ./lit_nlp/examples/gunicorn_config.py ./
+
+
+
+# ---- LIT Container for Hosted Demos ----
+
+FROM lit-nlp-base AS lit-nlp-prod
+
+RUN python -m pip install 'lit-nlp[examples-descai]'
+
+WORKDIR $APP_HOME
+ENTRYPOINT ["gunicorn", "--config=gunicorn_config.py"]
+
+
+
+# ---- LIT Container for Developing and Testing Hosted Demos ----
+
+FROM lit-nlp-base AS lit-nlp-dev
+
 # Install yarn
 RUN curl -sS https://dl.yarnpkg.com/debian/pubkey.gpg | apt-key add -
 RUN echo "deb https://dl.yarnpkg.com/debian/ stable main" | \
     tee /etc/apt/sources.list.d/yarn.list
 RUN apt update && apt -y install yarn
 
-# Copy local code to the container image.
-ENV APP_HOME=/app
-WORKDIR $APP_HOME
-
 # Set up python environment with production dependencies
 # This step is slow as it installs many packages.
-COPY ./requirements*.txt ./
-RUN python -m pip install -r requirements.txt
+COPY requirements.txt \
+     requirements_examples_common.txt \
+     requirements_examples_descai.txt \
+     ./
+RUN python -m pip install -r requirements_examples_descai.txt
 
 # Copy the rest of the lit_nlp package
 COPY . ./
@@ -47,7 +72,4 @@ RUN yarn && yarn build && rm -rf node_modules/*
 # Note that the config file supports configuring the LIT demo that is launched
 # via the DEMO_NAME and DEMO_PORT environment variables.
 WORKDIR $APP_HOME
-ENTRYPOINT [ \
-  "gunicorn", \
-  "--config=lit_nlp/examples/gunicorn_config.py" \
-]
+ENTRYPOINT ["gunicorn", "--config=gunicorn_config.py"]

lit_nlp/examples/gcp/Dockerfile

@@ -0,0 +1,173 @@
+# Copyright 2024 Google LLC
+#
+# Licensed under the Apache License, Version 2.0 (the "License");
+# you may not use this file except in compliance with the License.
+# You may obtain a copy of the License at
+#
+#     http://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing, software
+# distributed under the License is distributed on an "AS IS" BASIS,
+# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+# See the License for the specific language governing permissions and
+# limitations under the License.
+# ==============================================================================
+# Use the official lightweight Python image.
+# https://hub.docker.com/_/python
+
+# ---- LIT on GCP Base Images ----
+
+FROM python:3.11-slim AS lit-gcp-app-server-base
+
+# Update Ubuntu packages and install basic utils
+RUN apt-get update
+RUN apt-get install -y wget curl gnupg2 gcc g++ git
+
+# Copy local code to the container image.
+ENV APP_HOME=/app
+WORKDIR $APP_HOME
+
+COPY ./lit_nlp/examples/gcp/server_gunicorn_config.py ./gunicorn_config.py
+
+
+
+FROM nvidia/cuda:12.5.1-base-ubuntu22.04 AS lit-gcp-model-server-base
+ENV DEBIAN_FRONTEND=noninteractive
+ENV LANG=C.UTF-8
+
+ARG PYTHON_VERSION=python3.11
+
+RUN apt-get update
+
+# Install the CUDA Keyring package
+# See https://docs.nvidia.com/cuda/cuda-installation-guide-linux/index.html#network-repo-installation-for-ubuntu
+RUN apt-get install -y curl gnupg ca-certificates
+RUN curl https://developer.download.nvidia.com/compute/cuda/repos/ubuntu2204/x86_64/cuda-keyring_1.1-1_all.deb \
+    -o cuda-keyring_1.1-1_all.deb
+RUN dpkg -i cuda-keyring_1.1-1_all.deb
+
+# Install system and CUDA packages
+RUN apt-get install -y --no-install-recommends \
+    cuda-command-line-tools-12-3 \
+    cuda-cudart-dev-12-3 \
+    cuda-nvcc-12-3 \
+    cuda-cupti-12-3 \
+    cuda-nvprune-12-3 \
+    cuda-libraries-12-3 \
+    cuda-nvrtc-12-3 \
+    libcufft-12-3 \
+    libcurand-12-3 \
+    libcusolver-12-3 \
+    libcusparse-12-3 \
+    libcublas-12-3 \
+    libcudnn8=8.9.6.50-1+cuda12.2 \
+    libnvinfer-plugin8=8.6.1.6-1+cuda12.0 \
+    libnvinfer8=8.6.1.6-1+cuda12.0 \
+    build-essential \
+    pkg-config \
+    software-properties-common \
+    unzip
+
+# Install Python 3.11
+RUN apt-get install -y --no-install-recommends \
+    $PYTHON_VERSION \
+    $PYTHON_VERSION-venv \
+    $PYTHON_VERSION-distutils \
+    $PYTHON_VERSION-dev
+RUN ln -sf /usr/bin/$PYTHON_VERSION /usr/bin/python3
+RUN ln -sf /usr/bin/$PYTHON_VERSION /usr/bin/python
+
+# Install pip
+RUN curl https://bootstrap.pypa.io/get-pip.py -o get-pip.py
+RUN python3 get-pip.py
+RUN python3 -m pip install --no-cache-dir --upgrade pip
+
+RUN apt-get clean
+RUN rm -rf /var/lib/apt/lists/*
+
+COPY ./lit_nlp/examples/gcp/model_server_gunicorn_config.py ./
+
+
+
+# ---- LIT on GCP Production Images ----
+
+FROM lit-gcp-app-server-base AS lit-gcp-app-server
+
+RUN python -m pip install 'lit-nlp[examples-genai]'
+ENTRYPOINT ["gunicorn", "--config=gunicorn_config.py"]
+
+
+
+FROM lit-gcp-model-server-base AS lit-gcp-model-server
+
+RUN python -m pip install 'lit-nlp[examples-genai]'
+ENTRYPOINT ["gunicorn", "--config=model_server_gunicorn_config.py"]
+
+
+
+# ---- LIT on GCP Development Images ----
+
+FROM lit-gcp-app-server-base AS lit-gcp-app-server-dev
+
+# Install yarn
+RUN curl -sS https://dl.yarnpkg.com/debian/pubkey.gpg | apt-key add -
+RUN echo "deb https://dl.yarnpkg.com/debian/ stable main" | \
+    tee /etc/apt/sources.list.d/yarn.list
+RUN apt update && apt -y install yarn
+
+# Set up python environment with production dependencies
+# This step is slow as it installs many packages.
+COPY requirements.txt \
+     requirements_examples_common.txt \
+     requirements_examples_genai.txt \
+     ./
+RUN python -m pip install -r requirements_examples_genai.txt
+
+# Copy the rest of the lit_nlp package
+COPY . ./
+
+# Build front-end with yarn
+WORKDIR $APP_HOME/lit_nlp/client
+ENV NODE_OPTIONS="--openssl-legacy-provider"
+RUN yarn && yarn build && rm -rf node_modules/*
+
+# Run LIT server
+# Note that the config file supports configuring the LIT demo that is launched
+# via the DEMO_NAME and DEMO_PORT environment variables.
+WORKDIR $APP_HOME
+ENTRYPOINT ["gunicorn", "--config=gunicorn_config.py"]
+
+
+
+FROM lit-gcp-model-server-base AS lit-gcp-model-server-dev
+ENV APP_HOME=/app
+WORKDIR $APP_HOME
+
+# Install Node.js v18 (the base image ships with Node.js v12)
+# See https://github.com/nodesource/distributions
+RUN curl -fsSL https://deb.nodesource.com/setup_18.x -o nodesource_setup.sh
+RUN bash nodesource_setup.sh
+RUN apt-get install -y nodejs
+
+# Install yarn
+RUN curl -sS https://dl.yarnpkg.com/debian/pubkey.gpg | apt-key add -
+RUN echo "deb https://dl.yarnpkg.com/debian/ stable main" | \
+    tee /etc/apt/sources.list.d/yarn.list
+RUN apt update && apt -y install yarn
+
+COPY requirements.txt \
+     requirements_examples_common.txt \
+     requirements_examples_genai.txt \
+     ./
+RUN python -m pip install -r requirements_examples_genai.txt
+
+# Copy the rest of the lit_nlp package
+COPY . ./
+
+# Build front-end with yarn
+WORKDIR $APP_HOME/lit_nlp/client
+ENV NODE_OPTIONS="--openssl-legacy-provider"
+RUN yarn && yarn build && rm -rf node_modules/*
+
+WORKDIR $APP_HOME
+ENTRYPOINT ["gunicorn", "--config=model_server_gunicorn_config.py"]

lit_nlp/examples/gcp/README.md

@@ -0,0 +1,32 @@
+# Using LLMs in LIT on Google Cloud Platform
+
+## Developing
+
+### Use a virtual environment
+
+```shell
+# Create and activate the virtual environment
+python3 -m venv ~/.venvs/lit-on-gcp
+source ~/.venvs/lit-on-gcp/bin/activate
+
+# Install the requirements and LIT in editable mode
+pip install -f ./lit_nlp/examples/gcp/requirements.txt
+pip install -e .
+
+# Optionally, install tetsing requirements
+pip install -f ./requirements_test.txt
+pytest pytest lit_nlp/examples/gcp
+```
+
+### Build the Docker image
+
+```shell
+docker build -f ./lit_nlp/examples/gcp/Dockerfile -t lit-app:gcp-dev .
+```
+
+### Run GPT-2 in a Docker container
+
+```shell
+# Runs GPT-2 in Keras on PyTorch
+docker run --rm -p 5432:5432 -e MODEL_CONFIG=gpt2:gpt2_base_en lit-app:gcp-dev
+```

lit_nlp/examples/prompt_debugging/requirements.txt renamed to lit_nlp/examples/gcp/constants.py

@@ -1,4 +1,4 @@
-# Copyright 2023 Google LLC
+# Copyright 2024 Google LLC
 #
 # Licensed under the Apache License, Version 2.0 (the "License");
 # you may not use this file except in compliance with the License.
@@ -13,10 +13,14 @@
 # limitations under the License.
 # ==============================================================================
 
--r ../../../requirements_core.txt
+"""Constants used across the Model Server and LIT Server code surfaces."""
 
-keras>=3.1.0
-keras-nlp>=0.9.0
-tensorflow-text>=2.10.0
-torch>=2.0.0
-transformers>=4.27.1
+import enum
+
+
+class LlmHTTPEndpoints(enum.Enum):
+  """Names of HTTP endpoints provided by the Model Server conainer."""
+
+  GENERATE = 'predict'
+  SALIENCE = 'salience'
+  TOKENIZE = 'tokenize'