Add stop_token_ids in SamplingParams.__repr__ (#1745)

This commit is contained in:
陈序
2023-11-22 12:13:53 +08:00
committed by GitHub
parent 7d761fe3c1
commit 094f716bf2

View File

@ -241,6 +241,7 @@ class SamplingParams:
f"length_penalty={self.length_penalty}, "
f"early_stopping={self.early_stopping}, "
f"stop={self.stop}, "
f"stop_token_ids={self.stop_token_ids}, "
f"ignore_eos={self.ignore_eos}, "
f"max_tokens={self.max_tokens}, "
f"logprobs={self.logprobs}, "