From cdbda76fa95d062edbfb37295f43185303b1e867 Mon Sep 17 00:00:00 2001 From: Roy Han Date: Thu, 6 Jun 2024 16:17:07 -0700 Subject: [PATCH] Clean Up --- server/routes.go | 13 ++++--------- 1 file changed, 4 insertions(+), 9 deletions(-) diff --git a/server/routes.go b/server/routes.go index 7877e57b5..08e9bb60a 100644 --- a/server/routes.go +++ b/server/routes.go @@ -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 }