Skip to content

Commit

Permalink
fix linter nit
Browse files Browse the repository at this point in the history
  • Loading branch information
leloykun committed Aug 16, 2024
1 parent 146c6c0 commit e785a71
Showing 1 changed file with 3 additions and 1 deletion.
Original file line number Diff line number Diff line change
Expand Up @@ -150,7 +150,9 @@ def __call__(
)

# cast to desired return tensors type after concatenating
text_encoding = BatchEncoding(text_encoding, tensor_type=output_kwargs["common_kwargs"].get("return_tensors"))
text_encoding = BatchEncoding(
text_encoding, tensor_type=output_kwargs["common_kwargs"].get("return_tensors")
)
encoding.update(text_encoding)
qformer_text_encoding = self.qformer_tokenizer(text=text, **output_kwargs["text_kwargs"])
encoding["qformer_input_ids"] = qformer_text_encoding.pop("input_ids")
Expand Down

0 comments on commit e785a71

Please sign in to comment.