Spaces:
Paused
Paused
File size: 1,032 Bytes
4125443 5ae28aa 12fde71 5ae28aa 2c8b031 5ae28aa f5458b3 5ae28aa 2c8b031 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 |
FROM ghcr.io/ggerganov/llama.cpp:full-cuda
ENV DEBIAN_FRONTEND=noninteractive
# Update and install necessary dependencies
RUN apt update && \
apt install --no-install-recommends -y \
build-essential \
python3 \
python3-pip \
wget \
curl \
git \
cmake \
zlib1g-dev \
libblas-dev && \
apt clean && \
rm -rf /var/lib/apt/lists/*
# Setting up CUDA environment variables (this may not be necessary since you're using the official nvidia/cuda image, but it's good to be explicit)
ENV PATH="/usr/local/cuda/bin:$PATH" \
LD_LIBRARY_PATH="/usr/local/cuda/lib64:$LD_LIBRARY_PATH" \
CUDA_HOME="/usr/local/cuda"
WORKDIR /app
RUN wget https://huggingface.co/itsdotscience/Magicoder-S-DS-6.7B-GGUF/resolve/main/Magicoder-S-DS-6.7B_q8_0.gguf
RUN make LLAMA_CUBLAS=1
# Expose the port
EXPOSE 8080
# Start the llava-server with models
CMD ["--server", "--model", "Magicoder-S-DS-6.7B_q8_0.gguf", "--threads", "6", "--host", "0.0.0.0", "-ngl", "33"]
|