Skip to content

Commit 5c3f81a

Browse files
committed
upd tutorials
1 parent 1534056 commit 5c3f81a

File tree

2 files changed

+8
-7
lines changed

2 files changed

+8
-7
lines changed

user_guides/advanced/02_embedder_configuration.py

Lines changed: 5 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -29,7 +29,7 @@
2929
"""
3030

3131
# %%
32-
from autointent.configs import EmbedderConfig
32+
from autointent.configs import get_default_embedder_config
3333

3434
# Using a dictionary for detailed configuration
3535
advanced_embedder_config = {
@@ -59,7 +59,7 @@
5959

6060
from autointent.configs import TokenizerConfig
6161

62-
embedder_config = EmbedderConfig(
62+
embedder_config = get_default_embedder_config(
6363
model_name="sentence-transformers/all-mpnet-base-v2",
6464
batch_size=32,
6565
# Device is auto-detected, but you can override if needed
@@ -115,7 +115,7 @@
115115

116116
# %%
117117
# Example: Performance-optimized configuration
118-
perf_config = EmbedderConfig(
118+
perf_config = get_default_embedder_config(
119119
model_name="sentence-transformers/all-MiniLM-L6-v2", # Fast, lightweight model
120120
batch_size=128, # Large batch for speed
121121
# Device auto-detected by sentence-transformers
@@ -133,7 +133,7 @@
133133

134134
# %%
135135
# Example: Quality-optimized configuration
136-
quality_config = EmbedderConfig(
136+
quality_config = get_default_embedder_config(
137137
model_name="sentence-transformers/all-mpnet-base-v2", # High-quality model
138138
batch_size=16, # Smaller batch to handle longer sequences
139139
tokenizer_config=TokenizerConfig(max_length=512), # Longer sequences for context
@@ -151,7 +151,7 @@
151151

152152
# %%
153153
# Example: Multilingual setup
154-
multilingual_config = EmbedderConfig(
154+
multilingual_config = get_default_embedder_config(
155155
model_name="sentence-transformers/paraphrase-multilingual-MiniLM-L12-v2",
156156
batch_size=32,
157157
tokenizer_config=TokenizerConfig(max_length=256),

user_guides/basic_usage/03_automl.py

Lines changed: 3 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -115,10 +115,11 @@
115115
"""
116116

117117
# %%
118-
from autointent.configs import CrossEncoderConfig, EmbedderConfig, TokenizerConfig
118+
from autointent.configs import (CrossEncoderConfig, TokenizerConfig,
119+
get_default_embedder_config)
119120

120121
# Configure embedding model (used for vector representations)
121-
custom_pipeline.set_config(EmbedderConfig(model_name="prajjwal1/bert-tiny"))
122+
custom_pipeline.set_config(get_default_embedder_config(model_name="prajjwal1/bert-tiny"))
122123

123124
# Configure cross-encoder model (used for scoring text pairs)
124125
custom_pipeline.set_config(

0 commit comments

Comments
 (0)