Archives/langchain
2
0
Fork 1
mirror of https://github.com/hwchase17/langchain synced 2024-11-04 06:00:26 +00:00
Code Issues Packages Projects Releases Wiki Activity

cleanup

Browse Source
This commit is contained in:
olgavrou 2023-09-04 07:36:47 -04:00
parent ca163f0ee6
commit 67dc1a9dd2
2 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
libs/langchain/langchain/chains/rl_chain/__init__.py
View File

@ -15,6 +15,7 @@ from langchain.chains.rl_chain.base import (
from langchain.chains.rl_chain.pick_best_chain import (
PickBest,
PickBestEvent,
PickBestFeatureEmbedder,
PickBestSelected,
)
@ -37,6 +38,7 @@ __all__ = [
"PickBest",
"PickBestEvent",
"PickBestSelected",
"PickBestFeatureEmbedder",
"Embed",
"BasedOn",
"ToSelectFrom",

2
libs/langchain/langchain/chains/rl_chain/pick_best_chain.py
View File

@ -60,9 +60,7 @@ class PickBestFeatureEmbedder(base.Embedder[PickBestEvent]):
if model is None:
from sentence_transformers import SentenceTransformer
model = SentenceTransformer("all-mpnet-base-v2")
# model = SentenceTransformer("all-MiniLM-L6-v2")
self.model = model
self.auto_embed = auto_embed