[Minor] Rename no_eos_trim to no_stop_trim (#1661)

This commit is contained in:
Ying Sheng
2024-10-13 20:30:03 -07:00
committed by GitHub
parent da1ffed689
commit 2725f8da61
6 changed files with 16 additions and 16 deletions

View File

@@ -40,7 +40,7 @@ class SamplingParams:
regex: Optional[str] = None,
n: int = 1,
json_schema: Optional[str] = None,
no_eos_trim: bool = False,
no_stop_trim: bool = False,
) -> None:
self.temperature = temperature
self.top_p = top_p
@@ -61,7 +61,7 @@ class SamplingParams:
self.regex = regex
self.n = n
self.json_schema = json_schema
self.no_eos_trim = no_eos_trim
self.no_stop_trim = no_stop_trim
# Process some special cases
if self.temperature < _SAMPLING_EPS: