account for all non-repeating layers

This commit is contained in:
Michael Yang
2024-04-17 10:29:12 -07:00
parent 9df6c85c3a
commit a8b9b930b4
2 changed files with 50 additions and 11 deletions

View File

@@ -164,7 +164,8 @@ func (ts Tensors) Layers() map[string]Layer {
for _, t := range ts {
parts := strings.Split(t.Name, ".")
if parts[0] == "blk" {
parts = parts[1:]
// join first and second part, e.g. blk.%d
parts = append([]string{fmt.Sprintf("%s.%s", parts[0], parts[1])}, parts[2:]...)
}
if _, ok := layers[parts[0]]; !ok {