We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
Pool::Cls
JinaCodeBertModel
1 parent 6ce1559 commit ea24c2cCopy full SHA for ea24c2c
backends/candle/src/models/jina_code.rs
@@ -356,7 +356,9 @@ impl JinaCodeBertModel {
356
};
357
358
let pool = match model_type {
359
- ModelType::Classifier => Pool::Cls,
+ ModelType::Classifier => {
360
+ candle::bail!("`classifier` model type is not supported for JinaCode")
361
+ }
362
ModelType::Embedding(pool) => {
363
if pool == Pool::Splade {
364
candle::bail!("`splade` is not supported for JinaCode")
@@ -654,6 +656,7 @@ impl Model for JinaCodeBertModel {
654
656
fn is_padded(&self) -> bool {
655
657
true
658
}
659
+
660
fn embed(&self, batch: Batch) -> Result<(Option<Tensor>, Option<Tensor>)> {
661
self.forward(batch)
662
0 commit comments