Remove the useless to_srt_kwargs (#1955)
This commit is contained in:
@@ -133,17 +133,3 @@ class SamplingParams:
|
|||||||
else:
|
else:
|
||||||
stop_str_max_len = max(stop_str_max_len, len(stop_str))
|
stop_str_max_len = max(stop_str_max_len, len(stop_str))
|
||||||
self.stop_str_max_len = stop_str_max_len
|
self.stop_str_max_len = stop_str_max_len
|
||||||
|
|
||||||
def to_srt_kwargs(self):
|
|
||||||
return {
|
|
||||||
"max_new_tokens": self.max_new_tokens,
|
|
||||||
"stop": self.stop_strs,
|
|
||||||
"stop_token_ids": list(self.stop_token_ids),
|
|
||||||
"temperature": self.temperature,
|
|
||||||
"top_p": self.top_p,
|
|
||||||
"top_k": self.top_k,
|
|
||||||
"frequency_penalty": self.frequency_penalty,
|
|
||||||
"presence_penalty": self.presence_penalty,
|
|
||||||
"ignore_eos": self.ignore_eos,
|
|
||||||
"regex": self.regex,
|
|
||||||
}
|
|
||||||
|
|||||||
Reference in New Issue
Block a user