feature@init
This commit is contained in:
parent
8941872ed7
commit
1c6e8fd7fe
|
@ -4,7 +4,7 @@ from langchain.document_loaders import UnstructuredFileLoader
|
||||||
from langchain.embeddings.huggingface import HuggingFaceEmbeddings
|
from langchain.embeddings.huggingface import HuggingFaceEmbeddings
|
||||||
from langchain.vectorstores import FAISS
|
from langchain.vectorstores import FAISS
|
||||||
|
|
||||||
embedding_model_name = '/home/searchgpt/pretrained_models/ernie-gram-zh'
|
embedding_model_name = 'pretrained_models/ernie-gram-zh'
|
||||||
docs_path = 'docs'
|
docs_path = 'docs'
|
||||||
embeddings = HuggingFaceEmbeddings(model_name=embedding_model_name)
|
embeddings = HuggingFaceEmbeddings(model_name=embedding_model_name)
|
||||||
|
|
||||||
|
|
Loading…
Reference in New Issue