We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
1 parent 7cb4d93 commit 7072dbcCopy full SHA for 7072dbc
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='cuda')
29
_f('info', f'loading into {self.model.device}')
30
self.db = MilvusDB(self.config)
31
self.db.on()
0 commit comments