|
|
@@ -855,6 +855,7 @@ else:
|
|
|
default_text = settings['prompt']
|
|
|
description = f"\n\n# Text generation lab\nGenerate text using Large Language Models.\n"
|
|
|
css = ".my-4 {margin-top: 0} .py-6 {padding-top: 2.5rem} #refresh-button {flex: none; margin: 0; padding: 0; min-width: 50px; border: none; box-shadow: none; border-radius: 0} #download-label, #upload-label {min-height: 0}"
|
|
|
+chat_css = ".h-\[40vh\] {height: 66.67vh} .gradio-container {max-width: 800px; margin-left: auto; margin-right: auto} .w-screen {width: unset}"
|
|
|
suffix = '_pygmalion' if 'pygmalion' in model_name.lower() else ''
|
|
|
buttons = {}
|
|
|
gen_events = []
|
|
|
@@ -866,7 +867,7 @@ if args.chat or args.cai_chat:
|
|
|
if Path(f'logs/persistent.json').exists():
|
|
|
load_history(open(Path(f'logs/persistent.json'), 'rb').read(), settings[f'name1{suffix}'], settings[f'name2{suffix}'])
|
|
|
|
|
|
- with gr.Blocks(css=css+".h-\[40vh\] {height: 66.67vh} .gradio-container {max-width: 800px; margin-left: auto; margin-right: auto} .w-screen {width: unset}", analytics_enabled=False) as interface:
|
|
|
+ with gr.Blocks(css=css+chat_css, analytics_enabled=False) as interface:
|
|
|
if args.cai_chat:
|
|
|
display = gr.HTML(value=generate_chat_html(history['visible'], settings[f'name1{suffix}'], settings[f'name2{suffix}'], character))
|
|
|
else:
|
oobabooga