mirrors/ollama-for-amd
1
0
Fork 0
mirror of https://github.com/likelovewant/ollama-for-amd.git synced 2025-12-21 22:33:56 +00:00
Code Issues Packages Projects Releases Wiki Activity

bert

Browse Source
This commit is contained in:
Michael Yang
2024-06-06 08:59:04 -07:00
parent a017cf2fea
commit 5a28b9cf5f
6 changed files with 331 additions and 15 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
convert/reader.go
View File

@@ -37,6 +37,8 @@ const (
func (t tensorBase) Kind() uint32 {
if strings.HasSuffix(t.name, ".block_sparse_moe.gate.weight") {
return 0
} else if t.name == "embeddings.token_type_embeddings.weight" {
return 0
}
switch len(t.shape) {