Tomtom84 commited on
Commit
56c49c7
·
verified ·
1 Parent(s): 608fbe5

Update Dockerfile

Browse files
Files changed (1) hide show
  1. 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 ["uvicorn", "app:app", "--host", "0.0.0.0", "--port", "7860"]
 
 
 
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
+