Support for bert and nomic-bert embedding models

2025-12-11 00:07:07 +00:00 · 2024-02-20 21:37:29 -05:00
parent f0425d3de9
commit 63861f58cc
2 changed files with 14 additions and 0 deletions
--- a/server/images.go
+++ b/server/images.go
@@ -52,6 +52,10 @@ type Model struct {
 	Messages       []Message
 }

+func (m *Model) IsEmbedding() bool {
+	return slices.Contains(m.Config.ModelFamilies, "bert") || slices.Contains(m.Config.ModelFamilies, "nomic-bert")
+}
+
 type Message struct {
 	Role    string `json:"role"`
 	Content string `json:"content"`