saadkhi commited on
Commit
5977ece
·
verified ·
1 Parent(s): 9cc9857

Update Dockerfile

Browse files
Files changed (1) hide show
  1. Dockerfile +2 -2
Dockerfile CHANGED
@@ -2,7 +2,7 @@ FROM python:3.10-slim
2
 
3
  WORKDIR /app
4
 
5
- # System deps
6
  RUN apt-get update && apt-get install -y \
7
  ffmpeg \
8
  git \
@@ -13,7 +13,7 @@ RUN apt-get update && apt-get install -y \
13
  COPY requirements.txt .
14
  RUN pip install --no-cache-dir -r requirements.txt
15
 
16
- # Pre-download model (prevents runtime failures)
17
  RUN python -c "from transformers import AutoTokenizer, AutoModelForCausalLM; \
18
  AutoTokenizer.from_pretrained('TinyLlama/TinyLlama-1.1B-Chat-v1.0'); \
19
  AutoModelForCausalLM.from_pretrained('TinyLlama/TinyLlama-1.1B-Chat-v1.0')"
 
2
 
3
  WORKDIR /app
4
 
5
+ # System dependencies
6
  RUN apt-get update && apt-get install -y \
7
  ffmpeg \
8
  git \
 
13
  COPY requirements.txt .
14
  RUN pip install --no-cache-dir -r requirements.txt
15
 
16
+ # Pre-download model (avoids runtime network issues)
17
  RUN python -c "from transformers import AutoTokenizer, AutoModelForCausalLM; \
18
  AutoTokenizer.from_pretrained('TinyLlama/TinyLlama-1.1B-Chat-v1.0'); \
19
  AutoModelForCausalLM.from_pretrained('TinyLlama/TinyLlama-1.1B-Chat-v1.0')"