Skip to content

Commit 9342ba3

Browse files
committed
Use default Pool::Cls for {Jina,JinaCode}BertModel if classifier
1 parent c065def commit 9342ba3

File tree

2 files changed

+2
-6
lines changed

2 files changed

+2
-6
lines changed

backends/candle/src/models/jina.rs

Lines changed: 1 addition & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -367,9 +367,7 @@ impl JinaBertModel {
367367
};
368368

369369
let pool = match model_type {
370-
ModelType::Classifier => {
371-
candle::bail!("`classifier` model type is not supported for Jina")
372-
}
370+
ModelType::Classifier => Pool::Cls,
373371
ModelType::Embedding(pool) => {
374372
if pool == Pool::Splade {
375373
candle::bail!("`splade` is not supported for Jina")

backends/candle/src/models/jina_code.rs

Lines changed: 1 addition & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -356,9 +356,7 @@ impl JinaCodeBertModel {
356356
};
357357

358358
let pool = match model_type {
359-
ModelType::Classifier => {
360-
candle::bail!("`classifier` model type is not supported for JinaCode")
361-
}
359+
ModelType::Classifier => Pool::Cls,
362360
ModelType::Embedding(pool) => {
363361
if pool == Pool::Splade {
364362
candle::bail!("`splade` is not supported for JinaCode")

0 commit comments

Comments
 (0)