Skip to content

Commit 6679caa

Browse files
committed
fix: code format for inference.embedder.decoder_only.icl
1 parent 5d6fa9a commit 6679caa

1 file changed

Lines changed: 2 additions & 1 deletion

File tree

  • FlagEmbedding/inference/embedder/decoder_only

FlagEmbedding/inference/embedder/decoder_only/icl.py

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -343,7 +343,8 @@ def encode_queries_single_device(
343343
if device is None:
344344
device = self.target_devices[0]
345345

346-
if device == "cpu": self.model.float()
346+
if device == "cpu":
347+
self.model.float()
347348

348349
self.model.to(device)
349350
self.model.eval()

0 commit comments

Comments
 (0)