m97j's picture
Initial commit
d856372
# ----------- 베이슀 이미지 -----------
FROM python:3.10-slim
# ----------- μž‘μ—… 디렉토리 -----------
WORKDIR /app
# ----------- μ‹œμŠ€ν…œ νŒ¨ν‚€μ§€ μ„€μΉ˜ -----------
RUN apt-get update && apt-get install -y --no-install-recommends \
build-essential \
cmake \
&& rm -rf /var/lib/apt/lists/*
# ----------- μ˜μ‘΄μ„± μ„€μΉ˜ -----------
# requirements λ¨Όμ € 볡사 β†’ μΊμ‹œ ν™œμš© κ°€λŠ₯
COPY requirements.txt /app/requirements.txt
# PyTorch CPU 버전 μ„€μΉ˜ (GPU ν•„μš” μ—†μŒ)
RUN pip install --no-cache-dir torch==2.0.1+cpu -f https://download.pytorch.org/whl/cpu/torch_stable.html \
&& pip install --no-cache-dir -r /app/requirements.txt
# ----------- μ½”λ“œ 볡사 -----------
# μ˜μ‘΄μ„± μ„€μΉ˜ ν›„ μ½”λ“œ 볡사 β†’ requirements λ³€κ²½ μ—†λŠ” ν•œ μΊμ‹œ μž¬μ‚¬μš© κ°€λŠ₯
COPY . /app/
# ----------- κΆŒν•œ μ„€μ • 및 μ΄ˆκΈ°ν™” -----------
# κΆŒν•œ μ„€μ •
RUN chmod -R 777 /app
# 디렉토리 생성
RUN mkdir -p /app/models/fallback-npc-model \
&& mkdir -p /app/models/sentence-embedder \
&& mkdir -p /app/rag/chroma_DB \
&& mkdir -p /app/.cache/huggingface/hub \
mkdir -p /app/.cache/sentence_transformers && chmod -R 777 /app/.cache
# ν™˜κ²½λ³€μˆ˜ μ„€μ •
ENV TRANSFORMERS_CACHE=/app/.cache/huggingface/hub
ENV SENTENCE_TRANSFORMERS_HOME=/app/.cache/sentence_transformers
ENV FALLBACK_MODEL_NAME=skt/ko-gpt-trinity-1.2B-v0.5
ENV EMBEDDER_MODEL_NAME=sentence-transformers/paraphrase-multilingual-MiniLM-L12-v2
# ----------- 포트 μ„€μ • -----------
EXPOSE 7860
# ----------- μ‹€ν–‰ λͺ…λ Ή -----------
CMD ["uvicorn", "app:app", "--host", "0.0.0.0", "--port", "7860"]