Skip to content

Commit

Permalink
update device.
Browse files Browse the repository at this point in the history
  • Loading branch information
shibing624 committed Apr 16, 2023
1 parent 036dc58 commit dee54d2
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion text2vec/sentence_model.py
Original file line number Diff line number Diff line change
Expand Up @@ -141,7 +141,7 @@ def encode(
:param show_progress_bar: bool, Whether to show a progress bar for the sentences
:param convert_to_numpy: If true, the output is a list of numpy vectors. Else, it is a list of pytorch tensors.
:param convert_to_tensor: If true, you get one large tensor as return. Overwrites any setting from convert_to_numpy
:param device: Which torch.device to use for the computation
:param device: Which device to use for the computation
"""
self.bert.eval()
if device is None:
Expand Down

0 comments on commit dee54d2

Please sign in to comment.