File size: 1,633 Bytes
5fc69e4
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
0ce5a27
15092a9
 
 
b635cd2
15092a9
b635cd2
 
c04f910
 
b635cd2
c04f910
b635cd2
c04f910
216aedb
88d9111
 
d2afb53
5fc69e4
 
d856372
5fc69e4
 
d856372
a7eac01
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
# ----------- ๋ฒ ์ด์Šค ์ด๋ฏธ์ง€ -----------
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"]