github/langchain
1
0
Fork 0
mirror of https://github.com/hwchase17/langchain.git synced 2025-07-05 12:48:12 +00:00
Code Issues Packages Projects Releases Wiki Activity

community[patch]: fix top_p type hint (#15452)

Browse Source 
fix: https://github.com/langchain-ai/langchain/issues/15341

@efriis
This commit is contained in:
chyroc 2024-01-16 03:59:39 +08:00 committed by GitHub
parent 251afda549
commit d334efc848
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
4 changed files with 4 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
libs/community/langchain_community/embeddings/ollama.py
View File

@ -97,7 +97,7 @@ class OllamaEmbeddings(BaseModel, Embeddings):
will give more diverse answers, while a lower value (e.g. 10) will give more diverse answers, while a lower value (e.g. 10)
will be more conservative. (Default: 40)""" will be more conservative. (Default: 40)"""
top_p: Optional[int] = None top_p: Optional[float] = None
"""Works together with top-k. A higher value (e.g., 0.95) will lead """Works together with top-k. A higher value (e.g., 0.95) will lead
to more diverse text, while a lower value (e.g., 0.5) will to more diverse text, while a lower value (e.g., 0.5) will
generate more focused and conservative text. (Default: 0.9)""" generate more focused and conservative text. (Default: 0.9)"""

2
libs/community/langchain_community/llms/nlpcloud.py
View File

@ -38,7 +38,7 @@ class NLPCloud(LLM):
"""Whether or not to remove the end sequence token.""" """Whether or not to remove the end sequence token."""
bad_words: List[str] = [] bad_words: List[str] = []
"""List of tokens not allowed to be generated.""" """List of tokens not allowed to be generated."""
top_p: int = 1 top_p: float = 1.0
"""Total probability mass of tokens to consider at each step.""" """Total probability mass of tokens to consider at each step."""
top_k: int = 50 top_k: int = 50
"""The number of highest probability tokens to keep for top-k filtering.""" """The number of highest probability tokens to keep for top-k filtering."""

2
libs/community/langchain_community/llms/ollama.py
View File

@ -90,7 +90,7 @@ class _OllamaCommon(BaseLanguageModel):
will give more diverse answers, while a lower value (e.g. 10) will give more diverse answers, while a lower value (e.g. 10)
will be more conservative. (Default: 40)""" will be more conservative. (Default: 40)"""
top_p: Optional[int] = None top_p: Optional[float] = None
"""Works together with top-k. A higher value (e.g., 0.95) will lead """Works together with top-k. A higher value (e.g., 0.95) will lead
to more diverse text, while a lower value (e.g., 0.5) will to more diverse text, while a lower value (e.g., 0.5) will
generate more focused and conservative text. (Default: 0.9)""" generate more focused and conservative text. (Default: 0.9)"""

2
libs/partners/google-genai/langchain_google_genai/chat_models.py
View File

@ -443,7 +443,7 @@ Supported examples:
top_k: Optional[int] = None top_k: Optional[int] = None
"""Decode using top-k sampling: consider the set of top_k most probable tokens. """Decode using top-k sampling: consider the set of top_k most probable tokens.
Must be positive.""" Must be positive."""
top_p: Optional[int] = None top_p: Optional[float] = None
"""The maximum cumulative probability of tokens to consider when sampling. """The maximum cumulative probability of tokens to consider when sampling.
The model uses combined Top-k and nucleus sampling. The model uses combined Top-k and nucleus sampling.