theowise commited on
Commit
fa144ff
·
verified ·
1 Parent(s): 3602a6f

Update Dockerfile

Browse files
Files changed (1) hide show
  1. 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
- # 🌟 [핵심] 도커 이미지를 빌드할 때 임베딩 모델(442MB)을 리 다운로드하여 박아버립니다! (부팅시간 0초 매직)
18
- RUN python -c "from huggingface_hub import snapshot_download; snapshot_download(repo_id='jhgan/ko-sroberta-multitask')"
 
 
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 . .