Skip to content

GitLab

"vllm/vscode:/vscode.git/clone" did not exist on "ff36139ffc66294c19b503c1e52dc42c2cd265f6"
Unverified Commit 69909126 authored by sasha0552's avatar sasha0552 Committed by GitHub
Browse files

[Bugfix] Use random seed if seed is -1 (#4531)

parent e491c7e0
Show whitespace changes
Inline Side-by-side
Showing with 4 additions and 1 deletion
vllm/sampling_params.py
View file @ 69909126
...@@ -139,6 +139,9 @@ class SamplingParams: ...@@ -139,6 +139,9 @@ class SamplingParams:
self.top_p = top_p self.top_p = top_p
self.top_k = top_k self.top_k = top_k
self.min_p = min_p self.min_p = min_p
if seed == -1:
self.seed = None
else:
self.seed = seed self.seed = seed
self.use_beam_search = use_beam_search self.use_beam_search = use_beam_search
self.length_penalty = length_penalty self.length_penalty = length_penalty
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment