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 # optional
max_tokens=self.config['max_tokens'], max_tokens=self.config['max_tokens'],
n=1, n=1,
stop=None, stop=["\n###", "\n\n", "[PAD151645]", "<|im_end|>"],
top_p=self.config['top_p'], top_p=self.config['top_p'],
temperature=temperature, temperature=temperature,
frequency_penalty=self.config['frequency_penalty'], frequency_penalty=self.config['frequency_penalty'],