Update Dockerfile
Browse files- Dockerfile +4 -2
Dockerfile
CHANGED
|
@@ -14,8 +14,10 @@ WORKDIR /code
|
|
| 14 |
COPY requirements.txt .
|
| 15 |
RUN pip install --no-cache-dir -r requirements.txt
|
| 16 |
|
| 17 |
-
# 🌟 [핵심] 도커 이미지를 빌드할 때 임베딩 모델
|
| 18 |
-
RUN python -c "from huggingface_hub import snapshot_download;
|
|
|
|
|
|
|
| 19 |
|
| 20 |
# 5. 내 코드 전체 복사
|
| 21 |
COPY . .
|
|
|
|
| 14 |
COPY requirements.txt .
|
| 15 |
RUN pip install --no-cache-dir -r requirements.txt
|
| 16 |
|
| 17 |
+
# 🌟 [핵심] 도커 이미지를 빌드할 때 '1티어 임베딩 모델'과 '리랭커 모델'을 모두 다운로드하여 박아버립니다! (부팅시간 단축)
|
| 18 |
+
RUN python -c "from huggingface_hub import snapshot_download; \
|
| 19 |
+
snapshot_download(repo_id='BAAI/bge-m3'); \
|
| 20 |
+
snapshot_download(repo_id='Dongjin-kr/ko-reranker')"
|
| 21 |
|
| 22 |
# 5. 내 코드 전체 복사
|
| 23 |
COPY . .
|