Skip to content

Commit 221323b

Browse files
committed
fix: build issues
1 parent 5763cd3 commit 221323b

Some content is hidden

Large Commits have some content hidden by default. Use the searchbox below for content that may be hidden.

54 files changed

+458
-6002
lines changed

Cargo.lock

Lines changed: 0 additions & 15 deletions
Some generated files are not rendered by default. Learn more about customizing how changed files appear on GitHub.

Cargo.toml

Lines changed: 0 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -4,7 +4,6 @@ members = [
44
"backends/candle",
55
"backends/ort",
66
"backends/core",
7-
"backends/python",
87
"backends/grpc-client",
98
"core",
109
"router",
@@ -14,7 +13,6 @@ default-members = [
1413
"backends/candle",
1514
"backends/ort",
1615
"backends/core",
17-
"backends/python",
1816
"backends/grpc-client",
1917
"core",
2018
"router",

backends/Cargo.toml

Lines changed: 0 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -10,7 +10,6 @@ clap = { workspace = true, optional = true }
1010
hf-hub = { workspace = true }
1111
serde_json = { workspace = true }
1212
text-embeddings-backend-core = { path = "core" }
13-
text-embeddings-backend-python = { path = "python", optional = true }
1413
text-embeddings-backend-candle = { path = "candle", optional = true }
1514
text-embeddings-backend-ort = { path = "ort", optional = true }
1615
tokio = { workspace = true }
@@ -19,7 +18,6 @@ rand = { workspace = true }
1918

2019
[features]
2120
clap = ["dep:clap", "text-embeddings-backend-core/clap"]
22-
python = ["dep:text-embeddings-backend-python"]
2321
ort = ["dep:text-embeddings-backend-ort"]
2422
candle = ["dep:text-embeddings-backend-candle"]
2523
cuda = ["text-embeddings-backend-candle?/cuda"]

backends/candle/src/models/flash_bert.rs

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -259,6 +259,7 @@ impl FlashBertModel {
259259
};
260260
(pool, None, splade)
261261
}
262+
ModelType::ListwiseReranker => todo!(),
262263
};
263264

264265
let (embeddings, encoder) = match (
@@ -326,6 +327,7 @@ impl FlashBertModel {
326327
};
327328
(pool, None, splade)
328329
}
330+
ModelType::ListwiseReranker => todo!(),
329331
};
330332

331333
let (embeddings, encoder) = match (

backends/candle/src/models/flash_distilbert.rs

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -200,6 +200,7 @@ impl FlashDistilBertModel {
200200
candle::bail!("`classifier` model type is not supported for DistilBert")
201201
}
202202
ModelType::Embedding(pool) => pool,
203+
ModelType::ListwiseReranker => todo!(),
203204
};
204205

205206
let (embeddings, encoder) = match (

backends/candle/src/models/flash_gte.rs

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -191,6 +191,7 @@ impl FlashGTEModel {
191191
(pool, Some(classifier))
192192
}
193193
ModelType::Embedding(pool) => (pool, None),
194+
ModelType::ListwiseReranker => todo!(),
194195
};
195196

196197
let (word_embeddings, token_type_embeddings, layers, embeddings_norm) =

backends/candle/src/models/flash_jina.rs

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -267,6 +267,7 @@ impl FlashJinaBertModel {
267267
}
268268
(pool, None)
269269
}
270+
ModelType::ListwiseReranker => todo!(),
270271
};
271272

272273
let (embeddings, encoder) = match (

backends/candle/src/models/flash_jina_code.rs

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -314,6 +314,7 @@ impl FlashJinaCodeBertModel {
314314
}
315315
pool
316316
}
317+
ModelType::ListwiseReranker => todo!(),
317318
};
318319

319320
let (embeddings, encoder) = match (

backends/candle/src/models/flash_mistral.rs

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -251,6 +251,7 @@ impl FlashMistralModel {
251251
candle::bail!("`classifier` model type is not supported for Mistral")
252252
}
253253
ModelType::Embedding(pool) => pool,
254+
ModelType::ListwiseReranker => todo!(),
254255
};
255256

256257
let embeddings = Embedding::new(

backends/candle/src/models/flash_modernbert.rs

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -274,6 +274,7 @@ impl FlashModernBertModel {
274274

275275
(pool, None)
276276
}
277+
ModelType::ListwiseReranker => todo!(),
277278
};
278279

279280
let embeddings = ModernBertEmbeddings::load(vb.pp("model.embeddings"), config)

0 commit comments

Comments
 (0)