Spaces:
Paused
Paused
Update Dockerfile
Browse files- Dockerfile +5 -5
Dockerfile
CHANGED
|
@@ -26,15 +26,15 @@ RUN apt-get update && \
|
|
| 26 |
|
| 27 |
# Set the working directory
|
| 28 |
WORKDIR /app
|
| 29 |
-
|
| 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 |
|