drrobot9 commited on
Commit
8d2011b
·
verified ·
1 Parent(s): 4403e58

Update Dockerfile

Browse files
Files changed (1) hide show
  1. Dockerfile +3 -22
Dockerfile CHANGED
@@ -27,30 +27,11 @@ COPY requirements.txt .
27
  RUN pip install --no-cache-dir -r requirements.txt
28
 
29
 
30
- COPY yarngpt/ ./yarngpt/
31
 
32
 
33
- ENV PYTHONPATH="/app:${PYTHONPATH}"
34
-
35
- COPY app/ ./app/
36
-
37
-
38
- RUN mkdir -p /app/models && \
39
- python -c "import os; from huggingface_hub import snapshot_download; import time; print(' Downloading models from Remostartdev/TTS_FARMLINGUA-models...'); start = time.time(); snapshot_download(repo_id='Remostartdev/TTS_FARMLINGUA-models', local_dir='/app/models', local_dir_use_symlinks=False, resume_download=True, ignore_patterns=['*.git*', 'README.md']); print(f' Models downloaded in {time.time() - start:.1f}s')"
40
-
41
-
42
- RUN ls -la /app/models/ && \
43
- echo "Model files:" && \
44
- find /app/models -type f -name "*.ckpt" -o -name "*.yaml" | xargs -r ls -lh
45
-
46
- ENV MODEL_ID=saheedniyi/YarnGPT2b
47
- ENV WAV_TOKENIZER_CONFIG=/app/models/wavtokenizer_mediumdata_frame75_3s_nq1_code4096_dim512_kmeans200_attn.yaml
48
- ENV WAV_TOKENIZER_CKPT=/app/models/wavtokenizer_large_speech_320_24k.ckpt
49
-
50
-
51
- RUN python -c "from transformers import AutoModelForCausalLM; AutoModelForCausalLM.from_pretrained('saheedniyi/YarnGPT2b')"
52
 
53
  EXPOSE 7860
54
 
55
- # Gunicorn with Uvicorn workers for production
56
- CMD ["gunicorn", "app.main:app", "--workers", "4", "--worker-class", "uvicorn.workers.UvicornWorker", "--bind", "0.0.0.0:7860", "--timeout", "120", "--keep-alive", "5", "--max-requests", "10000", "--max-requests-jitter", "1000"]
 
27
  RUN pip install --no-cache-dir -r requirements.txt
28
 
29
 
30
+ COPY . .
31
 
32
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
33
 
34
  EXPOSE 7860
35
 
36
+
37
+ CMD ["uvicorn", "app.main:app", "--host", "0.0.0.0", "--port", "7860"]