diff --git a/FlagEmbedding/inference/embedder/encoder_only/base.py b/FlagEmbedding/inference/embedder/encoder_only/base.py index 2547ab24..0c759bca 100644 --- a/FlagEmbedding/inference/embedder/encoder_only/base.py +++ b/FlagEmbedding/inference/embedder/encoder_only/base.py @@ -74,12 +74,14 @@ def __init__( self.tokenizer = AutoTokenizer.from_pretrained( model_name_or_path, trust_remote_code=trust_remote_code, - cache_dir=cache_dir + cache_dir=cache_dir, + **kwargs ) self.model = AutoModel.from_pretrained( model_name_or_path, trust_remote_code=trust_remote_code, - cache_dir=cache_dir + cache_dir=cache_dir, + **kwargs ) def encode_queries(