From 996ceff08215f712fae32cae5684ffcc9d8d3d53 Mon Sep 17 00:00:00 2001 From: oobabooga <112222186+oobabooga@users.noreply.github.com> Date: Fri, 13 Jan 2023 11:32:12 -0300 Subject: [PATCH] Add PygmalionAI preset Borrowed from https://github.com/PygmalionAI/gradio-ui/blob/master/src/gradio_ui.py --- presets/PygmalionAI.txt | 7 +++++++ 1 file changed, 7 insertions(+) create mode 100644 presets/PygmalionAI.txt diff --git a/presets/PygmalionAI.txt b/presets/PygmalionAI.txt new file mode 100644 index 0000000..ef43098 --- /dev/null +++ b/presets/PygmalionAI.txt @@ -0,0 +1,7 @@ +do_sample=True, +max_new_tokens=max_length, +top_p=0.9, +top_k=0, +temperature=0.5, +repetition_penalty=1.05, +typical_p=1.0,