Skip to content

Commit 7072dbc

Browse files
committed
🚧 fix(wip): hardcode
1 parent 7cb4d93 commit 7072dbc

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

magnet/ize/memory.py

+1-1
Original file line numberDiff line numberDiff line change
@@ -25,7 +25,7 @@ class Embedder:
2525

2626
def __init__(self, config, create=False, initialize=False):
2727
self.config = config
28-
self.model = SentenceTransformer(self.config['MODEL'], device=Utils().check_cuda())
28+
self.model = SentenceTransformer(self.config['MODEL'], device='cuda')
2929
_f('info', f'loading into {self.model.device}')
3030
self.db = MilvusDB(self.config)
3131
self.db.on()

0 commit comments

Comments
 (0)