CooLLaMA / Dockerfile
CooLLaMACEO's picture
Update Dockerfile
81e3d0d verified
raw
history blame contribute delete
633 Bytes
FROM python:3.10-bullseye
# Install minimal dependencies for Llama
RUN apt-get update && apt-get install -y \
build-essential cmake libgomp1 wget \
&& rm -rf /var/lib/apt/lists/*
WORKDIR /app
# Upgrade pip + setuptools + wheel
RUN pip install --no-cache-dir --upgrade pip setuptools wheel
# Install llama-cpp-python
RUN pip install --no-cache-dir llama-cpp-python \
--extra-index-url https://abetlen.github.io/llama-cpp-python/whl/cpu
# Install Flask + Hugging Face tools
RUN pip install --no-cache-dir flask requests huggingface-hub
# Copy app
COPY . .
EXPOSE 7860
ENV PYTHONUNBUFFERED=1
CMD ["python", "app.py"]