File size: 964 Bytes
f1ec629
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
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
const_tag: &MODEL jukofyork/dark-miqu-70b

const_tag: &QK_ATTENUATION_FACTOR 0.8408964153  # sqrt(sqrt(1/2))
const_tag: &MLP_DOWN_SCALE_FACTOR 0.7071067812  # sqrt(1/2)

scale-filter-env: &scale_filter_env
  parameters:
    scale:
      - filter: q_proj
        value: *QK_ATTENUATION_FACTOR
      - filter: k_proj
        value: *QK_ATTENUATION_FACTOR
      - filter: down_proj
        value: *MLP_DOWN_SCALE_FACTOR
      - value: 1.0


slices:
  - sources:
    - model: *MODEL
      layer_range: [0, 20]
  - sources:
    - model: *MODEL
      layer_range: [20, 40]
      <<: *scale_filter_env
  - sources:
    - model: *MODEL
      layer_range: [20, 40]
      <<: *scale_filter_env
  - sources:
    - model: *MODEL
      layer_range: [40, 60]
      <<: *scale_filter_env
  - sources:
    - model: *MODEL
      layer_range: [40, 60]
      <<: *scale_filter_env
  - sources:
    - model: *MODEL
      layer_range: [60, 80]

merge_method: passthrough
dtype: float16