Update chatgpt-offline.py

This commit is contained in:
恍兮惚兮 2024-03-17 22:53:02 +08:00
parent 20084c4f43
commit 9caf26b083

View File

@ -47,7 +47,7 @@ class TS(basetrans):
# optional
max_tokens=self.config['max_tokens'],
n=1,
stop=None,
stop=["\n###", "\n\n", "[PAD151645]", "<|im_end|>"],
top_p=self.config['top_p'],
temperature=temperature,
frequency_penalty=self.config['frequency_penalty'],