Mikeplockhart commited on
Commit
361ac5d
·
verified ·
1 Parent(s): 69e6840

Update utils.py

Browse files
Files changed (1) hide show
  1. utils.py +2 -1
utils.py CHANGED
@@ -24,7 +24,8 @@ def embedding_function(items_to_embed: list[str]):
24
  )
25
  embedded_items = sentence_model.encode(
26
  items_to_embed,
27
- convert_to_numpy=False
 
28
  )
29
  print(len(embedded_items))
30
  print(type(embedded_items[0]))
 
24
  )
25
  embedded_items = sentence_model.encode(
26
  items_to_embed,
27
+ convert_to_numpy=False,
28
+ convert_to_tensor=False
29
  )
30
  print(len(embedded_items))
31
  print(type(embedded_items[0]))