File tree Expand file tree Collapse file tree 1 file changed +2
-2
lines changed
backends/python/server/text_embeddings_server/models Expand file tree Collapse file tree 1 file changed +2
-2
lines changed Original file line number Diff line number Diff 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 (
You can’t perform that action at this time.
0 commit comments