Spaces:
Runtime error
Runtime error
File size: 702 Bytes
e9cd1d1 f48b8b0 26155cf f48b8b0 fcf0970 f48b8b0 fcf0970 e9cd1d1 cb92f1e 51c1fe0 cb92f1e e9cd1d1 2500fa1 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 |
FROM python:3.9
WORKDIR /code
COPY ./requirements.txt /code/requirements.txt
RUN pip install --no-cache-dir --upgrade -r /code/requirements.txt
RUN apt-get update && apt-get install ffmpeg libsm6 libxext6 -y
# Create the cache directory and set permissions
RUN mkdir -p /code/transformers_cache && chmod -R 777 /code/transformers_cache
# Set the TRANSFORMERS_CACHE and HF_HOME environment variables
ENV TRANSFORMERS_CACHE=/code/transformers_cache
ENV HF_HOME=/code/transformers_cache
# Set PyTorch CUDA memory management variable
ENV PYTORCH_CUDA_ALLOC_CONF=garbage_collection_threshold:0.6,max_split_size_mb:128
COPY . .
CMD ["gunicorn", "-b", "0.0.0.0:7860", "main:app", "--timeout", "600"]
|