Add vectors argument to Model2Vec vectors, closes #846

neuml · Jan 7, 2025 · 2d8cf5d · 2d8cf5d
1 parent 3852043
commit 2d8cf5d
Showing 1 changed file with 5 additions and 1 deletion.
diff --git a/src/python/txtai/vectors/m2v.py b/src/python/txtai/vectors/m2v.py
@@ -59,4 +59,8 @@ def loadmodel(self, path):
         return StaticModel.from_pretrained(path)
 
     def encode(self, data):
-        return self.model.encode(data, batch_size=self.encodebatch)
+        # Additional model arguments
+        modelargs = self.config.get("vectors", {})
+
+        # Encode data
+        return self.model.encode(data, batch_size=self.encodebatch, **modelargs)