Skip to content

Commit a570b04

Browse files
ykyk
authored
Fixed worker requirements wrt transformers (LAION-AI#2621)
1 parent 95a6d53 commit a570b04

3 files changed

Lines changed: 1 addition & 3 deletions

File tree

‎docker/inference/Dockerfile.worker-hf‎

Lines changed: 0 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -20,7 +20,6 @@ RUN /conda/bin/conda init bash
2020
RUN /conda/bin/conda create -n oasst python=3.10 -y
2121

2222
COPY ./${APP_RELATIVE_PATH}/requirements.txt requirements-base.txt
23-
RUN grep -v "transformers" requirements-base.txt > requirements-base.txt.tmp && mv requirements-base.txt.tmp requirements-base.txt
2423
COPY ./${APP_RELATIVE_PATH}/requirements-hf.txt .
2524

2625
RUN cat requirements-base.txt requirements-hf.txt > requirements.txt

‎inference/worker/requirements-hf.txt‎

Lines changed: 0 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -4,5 +4,4 @@ fastapi
44
huggingface_hub
55
sse-starlette
66
torch
7-
git+https://github.com/huggingface/transformers@main#egg=transformers
87
uvicorn

‎inference/worker/requirements.txt‎

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -8,5 +8,5 @@ pydantic
88
requests
99
sentencepiece
1010
sseclient-py
11-
git+https://github.com/huggingface/transformers@60d51ef5123d949fd8c59cd4d3254e711541d278#egg=transformers
11+
git+https://github.com/huggingface/transformers@main#egg=transformers
1212
websocket-client

0 commit comments

Comments
 (0)