RafelSV commited on
Commit
4ddf3b0
·
verified ·
1 Parent(s): 0952513

Update Dockerfile

Browse files
Files changed (1) hide show
  1. Dockerfile +4 -2
Dockerfile CHANGED
@@ -10,7 +10,8 @@ WORKDIR /app
10
  COPY --chown=user requirements.txt requirements.txt
11
  RUN pip install --no-cache-dir --upgrade -r requirements.txt
12
 
13
- ENV IDEM_MODEL=QWEN05B
 
14
  ENV IDEM_LANG=es
15
  ENV IDEM_MAX_NEW_TOKENS=180
16
  ENV IDEM_MAX_INPUT_TOKENS=1024
@@ -20,8 +21,9 @@ ENV IDEM_MAX_CONTEXT_CHARS=2500
20
  ENV IDEM_CPU_THREADS=2
21
  ENV IDEM_CPU_SIMPLIFIER=true
22
 
23
- RUN python -c "from huggingface_hub import snapshot_download; snapshot_download('Qwen/Qwen2.5-0.5B-Instruct')"
24
 
25
  COPY --chown=user app.py app.py
 
26
 
27
  CMD ["uvicorn", "app:app", "--host", "0.0.0.0", "--port", "7860"]
 
10
  COPY --chown=user requirements.txt requirements.txt
11
  RUN pip install --no-cache-dir --upgrade -r requirements.txt
12
 
13
+ ENV IDEM_MODEL=LLAMA1B
14
+ ENV IDEM_USE_OFFICIAL_CORRECTIFIER=true
15
  ENV IDEM_LANG=es
16
  ENV IDEM_MAX_NEW_TOKENS=180
17
  ENV IDEM_MAX_INPUT_TOKENS=1024
 
21
  ENV IDEM_CPU_THREADS=2
22
  ENV IDEM_CPU_SIMPLIFIER=true
23
 
24
+ RUN python -c "from huggingface_hub import snapshot_download; snapshot_download('BSC-LT/salamandra-2b-instruct')"
25
 
26
  COPY --chown=user app.py app.py
27
+ COPY --chown=user core core
28
 
29
  CMD ["uvicorn", "app:app", "--host", "0.0.0.0", "--port", "7860"]