Spaces:
Sleeping
Sleeping
guangliang.yin
commited on
Commit
•
3d714c2
1
Parent(s):
23fff3a
优化代码-2
Browse files
project/embeddings/local_embed.py
CHANGED
@@ -6,7 +6,7 @@ from typing import Any, Dict, List, Optional
|
|
6 |
from langchain.embeddings.base import Embeddings
|
7 |
from langchain.pydantic_v1 import BaseModel, root_validator
|
8 |
from langchain.utils import get_from_dict_or_env
|
9 |
-
from FlagEmbedding import
|
10 |
|
11 |
logger = logging.getLogger(__name__)
|
12 |
|
@@ -44,7 +44,7 @@ class LocalEmbed(BaseModel, Embeddings):
|
|
44 |
"ZHIPUAI_API_KEY",
|
45 |
)
|
46 |
|
47 |
-
values["client"] =
|
48 |
query_instruction_for_retrieval="为这个句子生成表示以用于检索相关文章:",
|
49 |
use_fp16=True)
|
50 |
return values
|
|
|
6 |
from langchain.embeddings.base import Embeddings
|
7 |
from langchain.pydantic_v1 import BaseModel, root_validator
|
8 |
from langchain.utils import get_from_dict_or_env
|
9 |
+
from FlagEmbedding import FlagModel
|
10 |
|
11 |
logger = logging.getLogger(__name__)
|
12 |
|
|
|
44 |
"ZHIPUAI_API_KEY",
|
45 |
)
|
46 |
|
47 |
+
values["client"] = FlagModel('BAAI/bge-large-zh-v1.5',
|
48 |
query_instruction_for_retrieval="为这个句子生成表示以用于检索相关文章:",
|
49 |
use_fp16=True)
|
50 |
return values
|