Skip to content

Commit 0b4e081

Browse files
Refactor model initialization in example scripts to remove redundant model type specification
- Updated multiple example scripts to streamline the model initialization by removing the explicit model type (WhichModel) from the EmbeddingModel.from_pretrained_hf method calls. - This change enhances code readability and consistency across the examples.
1 parent 07fce35 commit 0b4e081

5 files changed

Lines changed: 6 additions & 6 deletions

File tree

examples/hybridsearch.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -47,15 +47,15 @@
4747

4848

4949
jina_model = EmbeddingModel.from_pretrained_hf(
50-
WhichModel.Jina, model_id="jinaai/jina-embeddings-v2-small-en"
50+
model_id="jinaai/jina-embeddings-v2-small-en"
5151
)
5252

5353
jina_embedddings = embed_anything.embed_query(sentences, embedder=jina_model)
5454
jina_query = embed_anything.embed_query(query_text, embedder=jina_model)[0]
5555

5656

5757
splade_model = EmbeddingModel.from_pretrained_hf(
58-
WhichModel.SparseBert, "prithivida/Splade_PP_en_v1"
58+
model_id="prithivida/Splade_PP_en_v1"
5959
)
6060
jina_embedddings = embed_anything.embed_query(sentences, embedder=jina_model)
6161

examples/s3_example.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -25,7 +25,7 @@
2525

2626
# Step 2: Embed the file
2727
embedder = EmbeddingModel.from_pretrained_hf(
28-
WhichModel.Jina, model_id="jinaai/jina-embeddings-v2-small-en"
28+
model_id="jinaai/jina-embeddings-v2-small-en"
2929
)
3030
embeddings = embedder.embed_file(file, config=TextEmbedConfig(chunk_size=1000, batch_size=32, buffer_size=64, splitting_strategy="sentence"))
3131
print(f"Embeddings: {embeddings}")

examples/splade.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -8,7 +8,7 @@
88

99

1010
model = EmbeddingModel.from_pretrained_hf(
11-
WhichModel.SparseBert, "prithivida/Splade_PP_en_v1"
11+
"prithivida/Splade_PP_en_v1"
1212
)
1313

1414
## ONNX model

examples/text.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -5,7 +5,7 @@
55

66
# Initialize the model once
77
model:EmbeddingModel = EmbeddingModel.from_pretrained_hf(
8-
WhichModel.Jina, model_id="jinaai/jina-embeddings-v2-small-en"
8+
model_id="jinaai/jina-embeddings-v2-small-en"
99
)
1010

1111

examples/text_ocr.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -7,7 +7,7 @@
77

88

99
model = EmbeddingModel.from_pretrained_hf(
10-
WhichModel.Jina, model_id="jinaai/jina-embeddings-v2-small-en"
10+
model_id="jinaai/jina-embeddings-v2-small-en"
1111
)
1212

1313
config = TextEmbedConfig(

0 commit comments

Comments
 (0)