chore: style

jina-ai · Sep 20, 2024 · c5e7ee8 · c5e7ee8
1 parent 99d9380
commit c5e7ee8
Show file tree

Hide file tree

Showing 2 changed files with 1 addition and 2 deletions.
diff --git a/chunked_pooling/wrappers.py b/chunked_pooling/wrappers.py
@@ -4,6 +4,7 @@
 import torch.nn as nn
 from transformers import AutoModel
 
+
 class JinaEmbeddingsV3Wrapper(nn.Module):
     def __init__(
         self, model_name, tasks=['retrieval.query', 'retrieval.passage'], **model_kwargs

diff --git a/run_chunked_eval.py b/run_chunked_eval.py
@@ -6,8 +6,6 @@
 from chunked_pooling.chunked_eval_tasks import *
 from chunked_pooling.wrappers import load_model
 
-from chunked_pooling.wrappers import load_model
-
 DEFAULT_CHUNKING_STRATEGY = 'fixed'
 DEFAULT_CHUNK_SIZE = 256
 DEFAULT_N_SENTENCES = 5