Update Dockerfile
Browse files- Dockerfile +1 -5
Dockerfile
CHANGED
@@ -37,11 +37,7 @@ RUN pip install --no-cache-dir -r requirements.txt
|
|
37 |
RUN python -c "from accelerate.utils import write_basic_config; write_basic_config(mixed_precision='fp16')" && \
|
38 |
chown -R appuser:appuser /app/.cache
|
39 |
|
40 |
-
|
41 |
-
RUN python -c "from transformers import AutoModelForCausalLM, AutoTokenizer; \
|
42 |
-
model = AutoModelForCausalLM.from_pretrained('gpt2-medium', use_safetensors=True); \
|
43 |
-
tokenizer = AutoTokenizer.from_pretrained('gpt2-medium')" && \
|
44 |
-
chown -R appuser:appuser /app/.cache
|
45 |
|
46 |
# Verify wkhtmltopdf installation
|
47 |
RUN which wkhtmltopdf && \
|
|
|
37 |
RUN python -c "from accelerate.utils import write_basic_config; write_basic_config(mixed_precision='fp16')" && \
|
38 |
chown -R appuser:appuser /app/.cache
|
39 |
|
40 |
+
|
|
|
|
|
|
|
|
|
41 |
|
42 |
# Verify wkhtmltopdf installation
|
43 |
RUN which wkhtmltopdf && \
|