style: ran linting

KennethEnevoldsen · Feb 17, 2024 · f729288 · f729288
1 parent 3b92090
commit f729288
Showing 1 changed file with 0 additions and 3 deletions.
diff --git a/src/seb/registered_models/e5_mistral.py b/src/seb/registered_models/e5_mistral.py
@@ -121,8 +121,6 @@ def preprocess(self, sentences: Sequence[str], instruction: str, encode_type: En
         ]
         batch_dict = self.tokenizer.pad(batch_dict, padding=True, return_attention_mask=True, return_tensors="pt")
 
-
-
         return batch_dict.to(self.model.device)
 
     # but it does not work slightly better than this:
@@ -157,7 +155,6 @@ def encode(
         else:
             instruction = ""
         for batch in tqdm(batched(sentences, batch_size)):
-
             with torch.inference_mode():
                 batch_dict = self.preprocess(batch, instruction=instruction, encode_type=encode_type)
                 outputs = self.model(**batch_dict)