ayush2917 commited on
Commit
463e55e
·
verified ·
1 Parent(s): abf566d

Update Dockerfile

Browse files
Files changed (1) hide show
  1. Dockerfile +2 -2
Dockerfile CHANGED
@@ -1,6 +1,6 @@
1
  FROM python:3.9-slim
2
 
3
- # Set environment variables (using HF_HOME instead of TRANSFORMERS_CACHE)
4
  ENV PYTHONUNBUFFERED=1 \
5
  HF_HOME=/tmp/cache \
6
  TORCH_HOME=/tmp/cache \
@@ -13,7 +13,7 @@ RUN mkdir -p /tmp/cache && \
13
  touch /tmp/news.db && \
14
  chmod -R 777 /tmp
15
 
16
- # Install system dependencies
17
  RUN apt-get update && \
18
  apt-get install -y --no-install-recommends \
19
  gcc \
 
1
  FROM python:3.9-slim
2
 
3
+ # Set environment variables
4
  ENV PYTHONUNBUFFERED=1 \
5
  HF_HOME=/tmp/cache \
6
  TORCH_HOME=/tmp/cache \
 
13
  touch /tmp/news.db && \
14
  chmod -R 777 /tmp
15
 
16
+ # Install system dependencies including accelerate
17
  RUN apt-get update && \
18
  apt-get install -y --no-install-recommends \
19
  gcc \