Commit 20733207 authored by Michael Yang's avatar Michael Yang Committed by Michael Yang
Browse files

fix lint

parent 93085127
...@@ -89,11 +89,11 @@ func (m *gptossModel) Tensors(ts []Tensor) []*ggml.Tensor { ...@@ -89,11 +89,11 @@ func (m *gptossModel) Tensors(ts []Tensor) []*ggml.Tensor {
out = append(out, slices.Collect(splitDim(t, 1, out = append(out, slices.Collect(splitDim(t, 1,
split{ split{
Replacer: strings.NewReplacer("gate_up_exps", "gate_exps"), Replacer: strings.NewReplacer("gate_up_exps", "gate_exps"),
slices: []tensor.Slice{nil, tensor.S(0, int(t.Shape()[1]), 2)}, slices: []tensor.Slice{nil, tensor.S(0, int(t.Shape()[1]), 2)},
}, },
split{ split{
Replacer: strings.NewReplacer("gate_up_exps", "up_exps"), Replacer: strings.NewReplacer("gate_up_exps", "up_exps"),
slices: []tensor.Slice{nil, tensor.S(1, int(t.Shape()[1]), 2)}, slices: []tensor.Slice{nil, tensor.S(1, int(t.Shape()[1]), 2)},
}, },
))...) ))...)
} else { } else {
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment