arindae commited on
Commit
8dc2d7a
·
1 Parent(s): 034506e

Update Dockerfile

Browse files
Files changed (1) hide show
  1. Dockerfile +4 -4
Dockerfile CHANGED
@@ -12,10 +12,10 @@ RUN pip install --no-cache-dir -r requirements.txt \
12
  RUN apt-get update && apt-get install -y --no-install-recommends patchelf \
13
  && find /usr/local/lib/python3.11/site-packages -name '*.so*' -path '*ctranslate2*' -exec patchelf --clear-execstack {} + \
14
  && apt-get purge -y patchelf && apt-get autoremove -y && rm -rf /var/lib/apt/lists/*
15
- RUN python convert_model.py # writes ./models/nllb-200-distilled-600M-int8
16
- RUN HF_MODEL=facebook/nllb-200-distilled-1.3B \
17
- CT2_MODEL_DIR=models/nllb-200-distilled-1.3B-int8 \
18
- python convert_model.py
19
  # MADLAD-400: fetch the pre-converted CTranslate2 int8 (model + tokenizer, ~3 GB)
20
  RUN python -c "from huggingface_hub import snapshot_download; snapshot_download('Nextcloud-AI/madlad400-3b-mt-ct2-int8', local_dir='models/madlad400-3b-mt-int8')"
21
 
 
12
  RUN apt-get update && apt-get install -y --no-install-recommends patchelf \
13
  && find /usr/local/lib/python3.11/site-packages -name '*.so*' -path '*ctranslate2*' -exec patchelf --clear-execstack {} + \
14
  && apt-get purge -y patchelf && apt-get autoremove -y && rm -rf /var/lib/apt/lists/*
15
+ # RUN python convert_model.py # writes ./models/nllb-200-distilled-600M-int8
16
+ # RUN HF_MODEL=facebook/nllb-200-distilled-1.3B \
17
+ # CT2_MODEL_DIR=models/nllb-200-distilled-1.3B-int8 \
18
+ # python convert_model.py
19
  # MADLAD-400: fetch the pre-converted CTranslate2 int8 (model + tokenizer, ~3 GB)
20
  RUN python -c "from huggingface_hub import snapshot_download; snapshot_download('Nextcloud-AI/madlad400-3b-mt-ct2-int8', local_dir='models/madlad400-3b-mt-int8')"
21