Fix Dockerfile pip install requirements file name.

PiperOrigin-RevId: 708384050
PAIR-code · Dec 20, 2024 · eb562e8 · eb562e8
1 parent c8c26b0
commit eb562e8
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 3 deletions.
diff --git a/Dockerfile b/Dockerfile
@@ -35,7 +35,7 @@ COPY ./lit_nlp/examples/gunicorn_config.py ./
 
 FROM lit-nlp-base AS lit-nlp-prod
 
-RUN python -m pip install 'lit-nlp[examples-descai]'
+RUN python -m pip install 'lit-nlp[examples-discriminative-ai]'
 
 WORKDIR $APP_HOME
 ENTRYPOINT ["gunicorn", "--config=gunicorn_config.py"]

diff --git a/lit_nlp/examples/gcp/Dockerfile b/lit_nlp/examples/gcp/Dockerfile
@@ -93,14 +93,14 @@ COPY ./lit_nlp/examples/gcp/model_server_gunicorn_config.py ./
 
 FROM lit-gcp-app-server-base AS lit-gcp-app-server
 
-RUN python -m pip install 'lit-nlp[examples-genai]'
+RUN python -m pip install 'lit-nlp[examples-generative-ai]'
 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]'
+RUN python -m pip install 'lit-nlp[examples-generative-ai]'
 ENTRYPOINT ["gunicorn", "--config=model_server_gunicorn_config.py"]