all repos — nixfiles @ e738ad2d1fc7405501fd22ebad27ad479c1bbd78

System and user configuration, managed by nix and home-manager

user/settings/llama/swap-config.yml (view raw)

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
37
38
39
40
41
42
43
44
45
46
47
macros:
  llama: |
    llama-server --port ${PORT}

  # - small `n` are not recommended
  # - MoEs require long drafts
  # - dense models: can reduce `--draft-min` and `--draft-max`
  next_edit: |
    --spec-type ngram-mod
    --spec-ngram-mod-n-match 24
    --spec-ngram-mod-n-min 24
    --spec-ngram-mod-n-max 32

models:
  sweep-next-edit-1.5b:
    name: Sweep Next-Edit 1.5B
    unlisted: true
    cmd: |
      ${llama} --hf-repo sweepai/sweep-next-edit-1.5B
      ${next_edit}
  sweep-next-edit-v2-7b:
    name: Sweep Next-Edit v2 7B
    unlisted: true
    cmd: |
      ${llama} --hf-repo henrik3/sweep-next-edit-v2-7B-GGUF
      ${next_edit}
  zeta2:
    name: Zeta 2
    unlisted: true
    cmd: |
      ${llama} --hf-repo bartowski/zed-industries_zeta-2-GGUF
      ${next_edit}

groups:
  edit_prediction:
    persistent: true # not unloaded by other groups
    swap: true # only run one model from this group
    exclusive: false
    members:
      - sweep-next-edit-1.5b
      - sweep-next-edit-v2-7b
      - zeta2

hooks:
  on_startup:
    preload:
      - sweep-next-edit-v2-7b