Skip to content

Commit

Permalink
add future readability
Browse files Browse the repository at this point in the history
  • Loading branch information
alperkaya committed Oct 7, 2024
1 parent a591489 commit d714f92
Showing 1 changed file with 3 additions and 1 deletion.
Original file line number Diff line number Diff line change
Expand Up @@ -104,12 +104,14 @@ def __init__(
local_files_only: bool = False,
model_kwargs: Optional[Dict[str, Any]] = None,
):
model_kwargs = model_kwargs or {}

self.model = SparseTextEmbedding(
model_name=model_name,
cache_dir=cache_dir,
threads=threads,
local_files_only=local_files_only,
**(model_kwargs if model_kwargs else {}),
**model_kwargs,
)

def embed(self, data: List[List[str]], progress_bar=True, **kwargs) -> List[SparseEmbedding]:
Expand Down

0 comments on commit d714f92

Please sign in to comment.