Skip to content

Commit af626ae

Browse files
committed
🦄 refactor: tiny refactor
1 parent 132b6cb commit af626ae

File tree

1 file changed

+0
-3
lines changed

1 file changed

+0
-3
lines changed

src/config/model.py

Lines changed: 0 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -34,16 +34,13 @@ class EmbeddingConfig(BaseConfig):
3434
class RerankerConfig(BaseConfig):
3535
backend: str
3636
batch_size: int
37-
max_seq_len: int
3837

3938
class SentenceTransformerRerankerConfig(RerankerConfig):
4039
backend: Literal["sentence_transformer"] = "sentence_transformer"
41-
batch_size: int
4240
max_seq_len: int
4341

4442

4543
class InfinityRerankerConfig(RerankerConfig):
4644
backend: Literal["infinity"] = "infinity"
4745
engine: Literal["torch"] = "torch"
48-
batch_size: int
4946
dtype: Literal["float32", "float16", "bfloat16"] = "float32"

0 commit comments

Comments
 (0)