File tree 4 files changed +40
-5
lines changed
4 files changed +40
-5
lines changed Original file line number Diff line number Diff line change 1
1
MODEL_DIR = /media/teamgroup/models
2
- MODEL = yarn-llama-2-13b-128k .Q5_K_M.gguf
2
+ MODEL = mistral-7b-instruct-v0.1 .Q5_K_M.gguf
Original file line number Diff line number Diff line change 1
1
# cuda devel image for base, best build compatibility
2
- FROM nvidia/cuda:11.8.0 -devel-ubuntu22.04 as builder
2
+ FROM nvidia/cuda:12.1.1 -devel-ubuntu22.04 as builder
3
3
4
4
# Update base image and install dependencies
5
5
RUN apt-get update && apt-get upgrade -y \
@@ -11,8 +11,10 @@ RUN apt-get update && apt-get upgrade -y \
11
11
12
12
WORKDIR /koboldcpp
13
13
14
+ ARG clone_arg
15
+
14
16
# Pulling latest koboldcpp branch and installing requirements
15
- RUN git clone https://github.com/LostRuins/koboldcpp.git --branch v1.44 ./
17
+ RUN git clone https://github.com/LostRuins/koboldcpp.git $clone_arg ./
16
18
17
19
RUN pip3 install -r requirements.txt
18
20
@@ -26,7 +28,7 @@ ENV LLAMA_OPENBLAS=1
26
28
RUN make
27
29
28
30
# Using runtime for smaller final image
29
- FROM nvidia/cuda:11.8.0 -runtime-ubuntu22.04
31
+ FROM noneabove1182/nvidia-runtime-docker:12.1.1 -runtime-ubuntu22.04
30
32
31
33
# update image and install necessary packages
32
34
RUN apt-get update && apt-get upgrade -y \
Original file line number Diff line number Diff line change @@ -34,5 +34,5 @@ services:
34
34
" --usecublas" ,
35
35
" 0" ,
36
36
" --gpulayers" ,
37
- " 22 "
37
+ " 100 "
38
38
]
Original file line number Diff line number Diff line change
1
+ #! /bin/bash
2
+
3
+ # Define the repository URL
4
+ REPO_URL=" https://api.github.com/repos/LostRuins/koboldcpp/releases/latest"
5
+
6
+ # Get the latest release tag from the GitHub repository
7
+ RELEASE_TAG=$( curl -s $REPO_URL | grep ' tag_name' | cut -d\" -f4)
8
+
9
+ # Define clone argument for Docker
10
+ if [ -z " $RELEASE_TAG " ]; then
11
+ echo " Failed to get the latest release tag. Exiting..."
12
+ exit 1
13
+ fi
14
+
15
+ echo $RELEASE_TAG
16
+
17
+ CLONE_ARG=" --branch $RELEASE_TAG "
18
+
19
+ # Build the Docker image
20
+ docker build --build-arg clone_arg=" $CLONE_ARG " -t noneabove1182/koboldcpp-gpu .
21
+
22
+ # Check if Docker build was successful
23
+ if [ $? -ne 0 ]; then
24
+ echo " Docker build failed. Exiting..."
25
+ exit 1
26
+ fi
27
+
28
+ # Tag the Docker image
29
+ docker tag noneabove1182/koboldcpp-gpu:latest noneabove1182/koboldcpp-gpu:$RELEASE_TAG
30
+
31
+ # Push the Docker images
32
+ docker push noneabove1182/koboldcpp-gpu:latest
33
+ docker push noneabove1182/koboldcpp-gpu:$RELEASE_TAG
You can’t perform that action at this time.
0 commit comments