credent007 commited on
Commit
2ff538a
·
verified ·
1 Parent(s): 8dc24e1

Update Dockerfile

Browse files
Files changed (1) hide show
  1. Dockerfile +5 -5
Dockerfile CHANGED
@@ -26,15 +26,15 @@ RUN apt-get update && \
26
 
27
  # Set the working directory
28
  WORKDIR /app
29
- RUN python3 -c "from transformers import AutoTokenizer, AutoModelForCausalLM; \
30
- model_name = 'microsoft/phi-3-mini-128k-instruct'; \
31
- AutoTokenizer.from_pretrained(model_name); \
32
- AutoModelForCausalLM.from_pretrained(model_name, trust_remote_code=True)"
33
  # Install Python dependencies first (leverages Docker layer caching)
34
  COPY requirements.txt .
35
  RUN pip install --no-cache-dir --upgrade pip && \
36
  pip install --no-cache-dir -r requirements.txt
37
-
 
 
 
38
  # Copy the rest of the application code
39
  COPY . .
40
 
 
26
 
27
  # Set the working directory
28
  WORKDIR /app
29
+
 
 
 
30
  # Install Python dependencies first (leverages Docker layer caching)
31
  COPY requirements.txt .
32
  RUN pip install --no-cache-dir --upgrade pip && \
33
  pip install --no-cache-dir -r requirements.txt
34
+ RUN python3 -c "from transformers import AutoTokenizer, AutoModelForCausalLM; \
35
+ model_name = 'microsoft/phi-3-mini-128k-instruct'; \
36
+ AutoTokenizer.from_pretrained(model_name); \
37
+ AutoModelForCausalLM.from_pretrained(model_name, trust_remote_code=True)"
38
  # Copy the rest of the application code
39
  COPY . .
40