hiddenFront commited on
Commit
dd0f82c
ยท
verified ยท
1 Parent(s): 8153817

Update Dockerfile

Browse files
Files changed (1) hide show
  1. Dockerfile +6 -2
Dockerfile CHANGED
@@ -25,10 +25,14 @@ RUN curl --proto '=https' --tlsv1.2 -sSf https://sh.rustup.rs | sh -s -- -y --de
25
  # rustup์€ ๊ธฐ๋ณธ์ ์œผ๋กœ /root/.cargo/bin์— ์„ค์น˜ํ•ฉ๋‹ˆ๋‹ค.
26
  ENV PATH="/root/.cargo/bin:${PATH}"
27
 
 
 
 
 
28
  # ์ž‘์—… ๋””๋ ‰ํ† ๋ฆฌ ์„ค์ •
29
  WORKDIR /app
30
 
31
- # ์ „์ฒด ์ฝ”๋“œ ๋ณต์‚ฌ (requirements.txt, main.py, .pkl ํŒŒ์ผ ๋“ฑ)
32
  COPY . .
33
 
34
  # Python ํŒจํ‚ค์ง€ ์„ค์น˜
@@ -50,7 +54,7 @@ RUN pip install --no-cache-dir \
50
  gluonnlp==0.10.0 \
51
  mxnet-mkl==1.6.0 \
52
  huggingface_hub \
53
- git+https://github.com/SKTBrain/KOBERT#egg=kobert_tokenizer\&subdirectory=kobert_hf # ์ด ๋ถ€๋ถ„์„ kobert_hf๋กœ ๋ณ€๊ฒฝํ–ˆ์Šต๋‹ˆ๋‹ค.
54
 
55
  # FastAPI ์• ํ”Œ๋ฆฌ์ผ€์ด์…˜ ์‹คํ–‰
56
  # --host 0.0.0.0: ๋ชจ๋“  ๋„คํŠธ์›Œํฌ ์ธํ„ฐํŽ˜์ด์Šค์—์„œ ์ ‘๊ทผ ํ—ˆ์šฉ
 
25
  # rustup์€ ๊ธฐ๋ณธ์ ์œผ๋กœ /root/.cargo/bin์— ์„ค์น˜ํ•ฉ๋‹ˆ๋‹ค.
26
  ENV PATH="/root/.cargo/bin:${PATH}"
27
 
28
+ # --- TRANSFORMERS_CACHE ํ™˜๊ฒฝ ๋ณ€์ˆ˜ ์„ค์ • ---
29
+ # Hugging Face ์บ์‹œ ๋””๋ ‰ํ† ๋ฆฌ๋ฅผ ์“ฐ๊ธฐ ๊ฐ€๋Šฅํ•œ /tmp ๋””๋ ‰ํ† ๋ฆฌ๋กœ ์„ค์ •ํ•ฉ๋‹ˆ๋‹ค.
30
+ ENV TRANSFORMERS_CACHE="/tmp/huggingface_cache"
31
+
32
  # ์ž‘์—… ๋””๋ ‰ํ† ๋ฆฌ ์„ค์ •
33
  WORKDIR /app
34
 
35
+ # ์ „์ฒด ์ฝ”๋“œ ๋ณต์‚ฌ (requirements.txt, app.py, .pkl ํŒŒ์ผ ๋“ฑ)
36
  COPY . .
37
 
38
  # Python ํŒจํ‚ค์ง€ ์„ค์น˜
 
54
  gluonnlp==0.10.0 \
55
  mxnet-mkl==1.6.0 \
56
  huggingface_hub \
57
+ git+https://github.com/SKTBrain/KOBERT#egg=kobert_tokenizer\&subdirectory=kobert_hf
58
 
59
  # FastAPI ์• ํ”Œ๋ฆฌ์ผ€์ด์…˜ ์‹คํ–‰
60
  # --host 0.0.0.0: ๋ชจ๋“  ๋„คํŠธ์›Œํฌ ์ธํ„ฐํŽ˜์ด์Šค์—์„œ ์ ‘๊ทผ ํ—ˆ์šฉ