mirror of
https://github.com/hwchase17/langchain.git
synced 2025-09-03 03:59:42 +00:00
community[patch]: fix top_p type hint (#15452)
fix: https://github.com/langchain-ai/langchain/issues/15341 @efriis
This commit is contained in:
@@ -38,7 +38,7 @@ class NLPCloud(LLM):
|
||||
"""Whether or not to remove the end sequence token."""
|
||||
bad_words: List[str] = []
|
||||
"""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."""
|
||||
top_k: int = 50
|
||||
"""The number of highest probability tokens to keep for top-k filtering."""
|
||||
|
Reference in New Issue
Block a user