fix huggingface embedder

rbiseck3 · rbiseck3 · commit db850a332118 · 2025-02-12T21:15:03.000-05:00
diff --git a/unstructured_ingest/embed/huggingface.py b/unstructured_ingest/embed/huggingface.py
@@ -15,9 +15,7 @@
 
 
 class HuggingFaceEmbeddingConfig(EmbeddingConfig):
-    embedder_model_name: Optional[str] = Field(
-        default="sentence-transformers/all-MiniLM-L6-v2", alias="model_name"
-    )
+    embedder_model_name: Optional[str] = Field(default="all-MiniLM-L6-v2", alias="model_name")
     embedder_model_kwargs: Optional[dict] = Field(
         default_factory=lambda: {"device": "cpu"}, alias="model_kwargs"
     )

Original file line number	Diff line number	Diff line change
`@@ -15,9 +15,7 @@`
`15`	`15`
`16`	`16`
`17`	`17`	`class HuggingFaceEmbeddingConfig(EmbeddingConfig):`
`18`		`- embedder_model_name: Optional[str] = Field(`
`19`		`- default="sentence-transformers/all-MiniLM-L6-v2", alias="model_name"`
`20`		`- )`
	`18`	`+ embedder_model_name: Optional[str] = Field(default="all-MiniLM-L6-v2", alias="model_name")`
`21`	`19`	`embedder_model_kwargs: Optional[dict] = Field(`
`22`	`20`	`default_factory=lambda: {"device": "cpu"}, alias="model_kwargs"`
`23`	`21`	`)`