third-party-mirrors/ollama
1
0
Fork 1
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

...

Browse Source
This commit is contained in:
Blake Mizerany 2024-04-17 17:04:13 -07:00
parent 0fbb379373
commit cfd4152eb6
1 changed files with 5 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
types/model/digest.go
View File

@ -1,6 +1,7 @@
package model package model
import ( import (
"fmt"
"log/slog" "log/slog"
"strings" "strings"
"unicode" "unicode"
@ -47,8 +48,11 @@ var (
// Digest. // Digest.
func ParseDigest(s string) Digest { func ParseDigest(s string) Digest {
typ, digest, ok := strings.Cut(s, "-") typ, digest, ok := strings.Cut(s, "-")
if !ok {
typ, digest, ok = strings.Cut(s, ":")
}
if ok && isValidDigestType(typ) && isValidHex(digest) { if ok && isValidDigestType(typ) && isValidHex(digest) {
return Digest{s: s} return Digest{s: fmt.Sprintf("%s-%s", typ, digest)}
} }
return Digest{} return Digest{}
} }