🚧 fix(wip):

Prismadic · Feb 23, 2024 · 7cb4d93 · 7cb4d93
1 parent 3f178ad
commit 7cb4d93
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/magnet/ize/memory.py b/magnet/ize/memory.py
@@ -25,7 +25,7 @@ class Embedder:
 
     def __init__(self, config, create=False, initialize=False):
         self.config = config
-        self.model = SentenceTransformer(self.config['MODEL'], device=Utils.check_cuda())
+        self.model = SentenceTransformer(self.config['MODEL'], device=Utils().check_cuda())
         _f('info', f'loading into {self.model.device}')
         self.db = MilvusDB(self.config)
         self.db.on()