Update Dockerfile
Browse files- Dockerfile +6 -2
Dockerfile
CHANGED
|
@@ -25,6 +25,10 @@ COPY --chown=user . /app
|
|
| 25 |
RUN pip install --no-cache-dir \
|
| 26 |
torch==2.3.1+cu121 torchaudio==2.3.1 \
|
| 27 |
--index-url https://download.pytorch.org/whl/cu121
|
| 28 |
-
|
|
|
|
|
|
|
| 29 |
EXPOSE 7860
|
| 30 |
-
CMD
|
|
|
|
|
|
|
|
|
| 25 |
RUN pip install --no-cache-dir \
|
| 26 |
torch==2.3.1+cu121 torchaudio==2.3.1 \
|
| 27 |
--index-url https://download.pytorch.org/whl/cu121
|
| 28 |
+
|
| 29 |
+
RUN pip install text-generation==0.6.1
|
| 30 |
+
ENV ORPHEUS_API_URL=http://127.0.0.1:1234/v1/completions
|
| 31 |
EXPOSE 7860
|
| 32 |
+
CMD bash -c "text-generation-launcher --model-id SebastianBodza/Kartoffel_Orpheus-3B_german_natural-v0.1 --port 1234 & \
|
| 33 |
+
uvicorn app:app --host 0.0.0.0 --port 7860"
|
| 34 |
+
|