Spaces:
Sleeping
Sleeping
Added cache setup
Browse files- Dockerfile +3 -1
Dockerfile
CHANGED
|
@@ -39,12 +39,14 @@ COPY models/ ./models/
|
|
| 39 |
COPY scripts/ ./scripts/
|
| 40 |
|
| 41 |
# Create cache directories with proper permissions
|
| 42 |
-
RUN mkdir -p /app/.cache/huggingface /tmp/numba_cache && \
|
| 43 |
chmod -R 777 /app/.cache /tmp/numba_cache
|
| 44 |
|
| 45 |
# Set environment
|
| 46 |
ENV PYTHONPATH="/app"
|
| 47 |
ENV HF_HOME="/app/.cache/huggingface"
|
|
|
|
|
|
|
| 48 |
ENV NUMBA_CACHE_DIR="/tmp/numba_cache"
|
| 49 |
ENV NUMBA_DISABLE_JIT=0
|
| 50 |
|
|
|
|
| 39 |
COPY scripts/ ./scripts/
|
| 40 |
|
| 41 |
# Create cache directories with proper permissions
|
| 42 |
+
RUN mkdir -p /app/.cache/huggingface /app/.cache/torch /tmp/numba_cache && \
|
| 43 |
chmod -R 777 /app/.cache /tmp/numba_cache
|
| 44 |
|
| 45 |
# Set environment
|
| 46 |
ENV PYTHONPATH="/app"
|
| 47 |
ENV HF_HOME="/app/.cache/huggingface"
|
| 48 |
+
ENV TRANSFORMERS_CACHE="/app/.cache/huggingface"
|
| 49 |
+
ENV TORCH_HOME="/app/.cache/torch"
|
| 50 |
ENV NUMBA_CACHE_DIR="/tmp/numba_cache"
|
| 51 |
ENV NUMBA_DISABLE_JIT=0
|
| 52 |
|