File size: 1,233 Bytes
9af25e9
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
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
dtype: bfloat16
merge_method: passthrough
# Layer-Interleaving of Yi-9B-200k
# where n = 48 (The number of layers the model originally had)
#       m = 12 (The number of layers to interleave per row)
#       M = 36 = 3m (The number of layers interleaved into the model)
#       s = 48 + 36 = n + M (The number of layers the model will have after the layer interleaving)
#       P = 15.75B parameters = (s/n)*9 (The number of parameters the model will have after the layer interleaving)
slices:
- sources:
  - layer_range: [0, 12]
    model: /Users/jsarnecki/opt/workspace/01-ai/Yi-9B-200K 

# Interleaved Layer 
- sources:
  - layer_range: [6, 18]
    model: /Users/jsarnecki/opt/workspace/01-ai/Yi-9B-200K

- sources:
  - layer_range: [12, 24]
    model: /Users/jsarnecki/opt/workspace/01-ai/Yi-9B-200K

# Interleaved Layer 
- sources:
  - layer_range: [18, 30]
    model: /Users/jsarnecki/opt/workspace/01-ai/Yi-9B-200K

- sources:
  - layer_range: [24, 36]
    model: /Users/jsarnecki/opt/workspace/01-ai/Yi-9B-200K
# Interleaved Layer 
- sources:
  - layer_range: [30, 42]
    model: /Users/jsarnecki/opt/workspace/01-ai/Yi-9B-200K

- sources:
  - layer_range: [36, 48]
    model: /Users/jsarnecki/opt/workspace/01-ai/Yi-9B-200K