This commit is contained in:
Roy Han 2024-06-06 16:17:07 -07:00
parent 30f7064363
commit cdbda76fa9

View File

@ -743,24 +743,19 @@ func getGGMLData(model *Model) ([]byte, error) {
f.Close()
kv := ggml.KV()
kvMap := make(map[string]any)
for k := range kv {
val := kv[k]
switch v := val.(type) {
switch v := kv[k].(type) {
case []interface{}:
if len(v) > 5 {
kvMap[k] = []string{}
kv[k] = []string{}
}
default:
kvMap[k] = val
}
}
kvMap["embedding_model"] = model.IsEmbedding()
kv["embedding_model"] = model.IsEmbedding()
ggmlJson, err := json.Marshal(kvMap)
ggmlJson, err := json.Marshal(kv)
if err != nil {
return nil, err
}