diff --git a/README.md b/README.md index 9efacb7c..65596321 100644 --- a/README.md +++ b/README.md @@ -187,8 +187,7 @@ For these two, please try commenting on an existing issue instead of creating a ## Credits +- Gradio dropdown menu refresh button: https://github.com/AUTOMATIC1111/stable-diffusion-webui +- Verbose preset: Anonymous 4chan user. - NovelAI and KoboldAI presets: https://github.com/KoboldAI/KoboldAI-Client/wiki/Settings-Presets - Pygmalion preset, code for early stopping in chat mode, code for some of the sliders, --chat mode colors: https://github.com/PygmalionAI/gradio-ui/ -- Verbose preset: Anonymous 4chan user. -- Instruct-Joi preset: https://huggingface.co/Rallio67/joi_12B_instruct_alpha -- Gradio dropdown menu refresh button: https://github.com/AUTOMATIC1111/stable-diffusion-webui diff --git a/presets/Contrastive Search.txt b/presets/Contrastive Search.txt new file mode 100644 index 00000000..832bc9ca --- /dev/null +++ b/presets/Contrastive Search.txt @@ -0,0 +1,3 @@ +do_sample=False +penalty_alpha=0.6 +top_k=4 diff --git a/presets/Instruct Joi (Contrastive Search).txt b/presets/Instruct Joi (Contrastive Search).txt deleted file mode 100644 index e17f0a66..00000000 --- a/presets/Instruct Joi (Contrastive Search).txt +++ /dev/null @@ -1,5 +0,0 @@ -top_p=0.95 -temperature=0.5 -penalty_alpha=0.6 -top_k=4 -repetition_penalty=1.03 diff --git a/server.py b/server.py index 9f584ba3..7d8792b7 100644 --- a/server.py +++ b/server.py @@ -197,11 +197,12 @@ shared.model, shared.tokenizer = load_model(shared.model_name) gen_events = [] default_preset = shared.settings['presets'][next((k for k in shared.settings['presets'] if re.match(k.lower(), shared.model_name.lower())), 'default')] default_text = shared.settings['prompts'][next((k for k in shared.settings['prompts'] if re.match(k.lower(), shared.model_name.lower())), 'default')] +title ='Text generation web UI' description = '\n\n# Text generation lab\nGenerate text using Large Language Models.\n' suffix = '_pygmalion' if 'pygmalion' in shared.model_name.lower() else '' if shared.args.chat or shared.args.cai_chat: - with gr.Blocks(css=ui.css+ui.chat_css, analytics_enabled=False) as shared.gradio['interface']: + with gr.Blocks(css=ui.css+ui.chat_css, analytics_enabled=False, title=title) as shared.gradio['interface']: if shared.args.cai_chat: shared.gradio['display'] = gr.HTML(value=generate_chat_html(shared.history['visible'], shared.settings[f'name1{suffix}'], shared.settings[f'name2{suffix}'], shared.character)) else: @@ -313,7 +314,7 @@ if shared.args.chat or shared.args.cai_chat: shared.gradio['interface'].load(reload_func, reload_inputs, [shared.gradio['display']], show_progress=True) elif shared.args.notebook: - with gr.Blocks(css=ui.css, analytics_enabled=False) as shared.gradio['interface']: + with gr.Blocks(css=ui.css, analytics_enabled=False, title=title) as shared.gradio['interface']: gr.Markdown(description) with gr.Tab('Raw'): shared.gradio['textbox'] = gr.Textbox(value=default_text, lines=23) @@ -337,7 +338,7 @@ elif shared.args.notebook: shared.gradio['Stop'].click(None, None, None, cancels=gen_events) else: - with gr.Blocks(css=ui.css, analytics_enabled=False) as shared.gradio['interface']: + with gr.Blocks(css=ui.css, analytics_enabled=False, title=title) as shared.gradio['interface']: gr.Markdown(description) with gr.Row(): with gr.Column():