itod commited on
Commit
8886c5e
·
verified ·
1 Parent(s): ff93ce0

make a writable cache

Browse files
Files changed (1) hide show
  1. Dockerfile +4 -0
Dockerfile CHANGED
@@ -7,6 +7,10 @@ ARG MODEL_NAME=WhereIsAI/UAE-Large-V1
7
  # Set the model name and path as environment variables
8
  ENV MODEL_NAME=$MODEL_NAME
9
  ENV MODEL_PATH=/models/${MODEL_NAME}
 
 
 
 
10
 
11
  # Install necessary Python packages
12
  COPY requirements.txt /app/requirements.txt
 
7
  # Set the model name and path as environment variables
8
  ENV MODEL_NAME=$MODEL_NAME
9
  ENV MODEL_PATH=/models/${MODEL_NAME}
10
+ ENV TRANSFORMERS_CACHE=/.cache
11
+
12
+ # Create /.cache directory and make it writable
13
+ RUN mkdir /.cache && chmod 777 /.cache
14
 
15
  # Install necessary Python packages
16
  COPY requirements.txt /app/requirements.txt