vbatts/llama.cpp
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

disable

Browse source
This commit is contained in:
Robert Washbourne 2023-11-24 01:56:36 -05:00
parent 7280bb217a
commit 1cde43fbb4
1 changed files with 3 additions and 3 deletions
Download patch file Download diff file Expand all files Collapse all files

6
.devops/main-cuda.Dockerfile
View file

@ -23,11 +23,11 @@ COPY . .
ENV CUDA_DOCKER_ARCH=${CUDA_DOCKER_ARCH} ENV CUDA_DOCKER_ARCH=${CUDA_DOCKER_ARCH}
# Enable cuBLAS # Enable cuBLAS
ENV LLAMA_CUBLAS=1 ENV LLAMA_CUBLAS=1
ENV LLAMA_CUDA_MMV_Y=2 # ENV LLAMA_CUDA_MMV_Y=2
ENV LLAMA_CUDA_DMMV_X=64 # ENV LLAMA_CUDA_DMMV_X=64
ENV LLAMA_CUDA_F16=true ENV LLAMA_CUDA_F16=true
RUN make RUN make -j
# Accept the build argument into an environment variable # Accept the build argument into an environment variable
ARG MODEL_URL ARG MODEL_URL