Skip to content

Commit 8016c4f

Browse files
committed
fix: correct XProvence class name (XProvence, not XProvenceForSequenceClassification)
1 parent 09e8491 commit 8016c4f

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

backends/python/server/text_embeddings_server/models/xprovence_model.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -76,7 +76,7 @@ def __init__(
7676

7777
# Get the custom model class directly from the dynamic module
7878
model_class = get_class_from_dynamic_module(
79-
"modeling_xprovence_hf.XProvenceForSequenceClassification",
79+
"modeling_xprovence_hf.XProvence",
8080
model_id,
8181
cache_dir=cache_dir,
8282
)
@@ -91,7 +91,7 @@ def __init__(
9191
# Fallback for local paths - try to import from local path
9292
logger.info(f"XProvence: Loading from local path {model_path}")
9393
model_class = get_class_from_dynamic_module(
94-
"modeling_xprovence_hf.XProvenceForSequenceClassification",
94+
"modeling_xprovence_hf.XProvence",
9595
model_path,
9696
)
9797
model = model_class.from_pretrained(

0 commit comments

Comments
 (0)