Clean Up
This commit is contained in:
parent
30f7064363
commit
cdbda76fa9
@ -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
|
||||||
}
|
}
|
||||||
|
Loading…
x
Reference in New Issue
Block a user