Skip to content

Commit

Permalink
feat: add codestral (#253)
Browse files Browse the repository at this point in the history
Signed-off-by: Sertac Ozercan <sozercan@gmail.com>
  • Loading branch information
sozercan authored May 30, 2024
1 parent dd09266 commit 2573ff3
Show file tree
Hide file tree
Showing 3 changed files with 33 additions and 0 deletions.
1 change: 1 addition & 0 deletions .github/workflows/patch-models.yaml
Original file line number Diff line number Diff line change
Expand Up @@ -25,6 +25,7 @@ jobs:
- ghcr.io/sozercan/mixtral:8x7b
- ghcr.io/sozercan/phi3:3.8b
- ghcr.io/sozercan/gemma1.1:2b
- ghcr.io/sozercan/codestral0.1:22b
steps:
- uses: jlumbroso/free-disk-space@54081f138730dfa15788a46383842cd2f914a1be # v1.3.1
with:
Expand Down
1 change: 1 addition & 0 deletions .github/workflows/update-models.yaml
Original file line number Diff line number Diff line change
Expand Up @@ -25,6 +25,7 @@ jobs:
- llama-3-8b-instruct
- phi-3-3.8b
- gemma-1.1-2b-instruct
- codestral-0.1-22b
runs-on: ubuntu-latest
timeout-minutes: 360
steps:
Expand Down
31 changes: 31 additions & 0 deletions models/codestral-0.1-22b.yaml
Original file line number Diff line number Diff line change
@@ -0,0 +1,31 @@
#syntax=ghcr.io/sozercan/aikit:latest
apiVersion: v1alpha1
debug: true
runtime: cuda
models:
- name: codestral-0.1-22b
source: https://huggingface.co/lmstudio-community/Codestral-22B-v0.1-GGUF/resolve/main/Codestral-22B-v0.1-Q4_K_M.gguf
sha256: defc9e0a1bb42857558d43df4e7f0f3d0a29d06a953e498e967d763f45d10431
promptTemplates:
- name: instruct
template: |
[INST]{{ if .SystemPrompt }}{{ .SystemPrompt }}{{ end }} {{ .Input }}[/INST]
config: |
- name: codestral-0.1-22b
backend: llama
parameters:
model: Codestral-22B-v0.1-Q4_K_M.gguf
context_size: 8192
template:
chat: instruct
completion: instruct
stopwords:
- \"[INST]\"
- \"[/INST]\"
- \"[PREFIX]\"
- \"[MIDDLE]\"
- \"[SUFFIX]\"
gpu_layers: 50
f16: true
batch: 512
mmap: true

0 comments on commit 2573ff3

Please sign in to comment.