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() f.Close()
kv := ggml.KV() kv := ggml.KV()
kvMap := make(map[string]any)
for k := range kv { for k := range kv {
val := kv[k] switch v := kv[k].(type) {
switch v := val.(type) {
case []interface{}: case []interface{}:
if len(v) > 5 { 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 { if err != nil {
return nil, err return nil, err
} }