Soumik-404 commited on
Commit
f71c424
·
1 Parent(s): 27bdca7
Files changed (1) hide show
  1. Dockerfile +1 -13
Dockerfile CHANGED
@@ -24,19 +24,7 @@ RUN pip install --no-cache-dir "youtube-transcript-api>=1.2.4"
24
 
25
  COPY --chown=appuser:appuser . .
26
 
27
- RUN mkdir -p /app/models && python3 -c "
28
- import json, os
29
- from huggingface_hub import snapshot_download
30
- snapshot_download(repo_id='ibm-granite/granite-embedding-small-english-r2', local_dir='/app/models/bge-384')
31
- snapshot_download(repo_id='nomic-ai/modernbert-embed-base', local_dir='/app/models/bge-768')
32
- snapshot_download(repo_id='lightonai/modernbert-embed-large', local_dir='/app/models/bge-1024')
33
- snapshot_download(repo_id='nomic-ai/nomic-embed-vision-v1.5', local_dir='/app/models/vision')
34
- cfg = '/app/models/vision/config.json'
35
- if os.path.exists(cfg):
36
- with open(cfg) as f: d = json.load(f)
37
- if isinstance(d.get('n_inner'), float): d['n_inner'] = int(d['n_inner'])
38
- with open(cfg, 'w') as f: json.dump(d, f, indent=2)
39
- " && chown -R appuser:appuser /app/models
40
 
41
  RUN mkdir -p /app/logs && \
42
  chown -R appuser:appuser /app/logs
 
24
 
25
  COPY --chown=appuser:appuser . .
26
 
27
+ RUN mkdir -p /app/models && python3 -c "from huggingface_hub import snapshot_download; snapshot_download(repo_id='ibm-granite/granite-embedding-small-english-r2', local_dir='/app/models/bge-384'); snapshot_download(repo_id='nomic-ai/modernbert-embed-base', local_dir='/app/models/bge-768'); snapshot_download(repo_id='lightonai/modernbert-embed-large', local_dir='/app/models/bge-1024'); snapshot_download(repo_id='nomic-ai/nomic-embed-vision-v1.5', local_dir='/app/models/vision'); import json, os; cfg='/app/models/vision/config.json'; d=json.load(open(cfg)); d['n_inner']=int(d['n_inner']) if isinstance(d.get('n_inner'),float) else d['n_inner']; json.dump(d, open(cfg,'w'), indent=2)" && chown -R appuser:appuser /app/models
 
 
 
 
 
 
 
 
 
 
 
 
28
 
29
  RUN mkdir -p /app/logs && \
30
  chown -R appuser:appuser /app/logs