Skip to content

Commit 529d9e1

Browse files
committed
🚧 fix(wip): debug device
1 parent 4f26105 commit 529d9e1

File tree

1 file changed

+1
-0
lines changed

1 file changed

+1
-0
lines changed

magnet/ize/memory.py

+1
Original file line numberDiff line numberDiff line change
@@ -25,6 +25,7 @@ class Embedder:
2525
def __init__(self, config, create=False, initialize=False):
2626
self.config = config
2727
self.model = SentenceTransformer(self.config['MODEL'])
28+
_f('info', f'loading into {self.model.device}')
2829
self.db = MilvusDB(self.config)
2930
self.db.on()
3031
if create:

0 commit comments

Comments
 (0)