We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
1 parent 3f178ad commit 7cb4d93Copy full SHA for 7cb4d93
magnet/ize/memory.py
@@ -25,7 +25,7 @@ class Embedder:
25
26
def __init__(self, config, create=False, initialize=False):
27
self.config = config
28
- self.model = SentenceTransformer(self.config['MODEL'], device=Utils.check_cuda())
+ self.model = SentenceTransformer(self.config['MODEL'], device=Utils().check_cuda())
29
_f('info', f'loading into {self.model.device}')
30
self.db = MilvusDB(self.config)
31
self.db.on()
0 commit comments