mike23415 commited on
Commit
02d5064
·
verified ·
1 Parent(s): 7905cfc

Update Dockerfile

Browse files
Files changed (1) hide show
  1. 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
- # Pre-download model (GPT-2 Medium)
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 && \