AIencoder commited on
Commit
adbd890
·
verified ·
1 Parent(s): 365df26

Update Dockerfile

Browse files
Files changed (1) hide show
  1. Dockerfile +13 -0
Dockerfile CHANGED
@@ -26,6 +26,19 @@ RUN pip install --no-cache-dir \
26
  faster-whisper \
27
  huggingface_hub
28
 
 
 
 
 
 
 
 
 
 
 
 
 
 
29
  # Copy application
30
  COPY app.py /app/app.py
31
 
 
26
  faster-whisper \
27
  huggingface_hub
28
 
29
+ # Pre-download all models during build
30
+ RUN mkdir -p /data/models && python -c "\
31
+ from huggingface_hub import hf_hub_download; \
32
+ models = [\
33
+ ('bartowski/DeepSeek-Coder-V2-Lite-Instruct-GGUF', 'DeepSeek-Coder-V2-Lite-Instruct-Q4_K_M.gguf'),\
34
+ ('Qwen/Qwen2.5-Coder-7B-Instruct-GGUF', 'qwen2.5-coder-7b-instruct-q4_k_m.gguf'),\
35
+ ('Qwen/Qwen2.5-Coder-3B-Instruct-GGUF', 'qwen2.5-coder-3b-instruct-q4_k_m.gguf'),\
36
+ ('Qwen/Qwen2.5-Coder-1.5B-Instruct-GGUF', 'qwen2.5-coder-1.5b-instruct-q4_k_m.gguf'),\
37
+ ('Qwen/Qwen2.5-Coder-0.5B-Instruct-GGUF', 'qwen2.5-coder-0.5b-instruct-q4_k_m.gguf'),\
38
+ ]; \
39
+ [print(f'Downloading {f}...') or hf_hub_download(repo_id=r, filename=f, local_dir='/data/models', local_dir_use_symlinks=False) for r, f in models]; \
40
+ print('All models downloaded!')"
41
+
42
  # Copy application
43
  COPY app.py /app/app.py
44