File tree Expand file tree Collapse file tree 2 files changed +2
-6
lines changed
backends/candle/src/models Expand file tree Collapse file tree 2 files changed +2
-6
lines changed Original file line number Diff line number Diff line change @@ -367,9 +367,7 @@ impl JinaBertModel {
367
367
} ;
368
368
369
369
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 ,
373
371
ModelType :: Embedding ( pool) => {
374
372
if pool == Pool :: Splade {
375
373
candle:: bail!( "`splade` is not supported for Jina" )
Original file line number Diff line number Diff line change @@ -356,9 +356,7 @@ impl JinaCodeBertModel {
356
356
} ;
357
357
358
358
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 ,
362
360
ModelType :: Embedding ( pool) => {
363
361
if pool == Pool :: Splade {
364
362
candle:: bail!( "`splade` is not supported for JinaCode" )
You can’t perform that action at this time.
0 commit comments