diff --git a/modules/shared.py b/modules/shared.py index d59cee99..f0408068 100644 --- a/modules/shared.py +++ b/modules/shared.py @@ -39,6 +39,8 @@ settings = { 'name2_pygmalion': 'Kawaii', 'context_pygmalion': "Kawaii's persona: Kawaii is a cheerful person who loves to make others smile. She is an optimist who loves to spread happiness and positivity wherever she goes.\n", 'stop_at_newline_pygmalion': False, + 'default_extensions': [], + 'chat_default_extensions': ["gallery"], } parser = argparse.ArgumentParser(formatter_class=lambda prog: argparse.HelpFormatter(prog,max_help_position=54)) diff --git a/server.py b/server.py index f64b526d..564c8644 100644 --- a/server.py +++ b/server.py @@ -151,7 +151,18 @@ available_characters = get_available_characters() available_softprompts = get_available_softprompts() extensions_module.available_extensions = get_available_extensions() -if shared.args.extensions is not None: +# Activate the default extensions +if shared.args.chat or shared.args.cai_chat: + for extension in shared.settings['chat_default_extensions']: + shared.args.extensions = shared.args.extensions or [] + if extension not in shared.args.extensions: + shared.args.extensions.append(extension) +else: + for extension in shared.settings['default_extensions']: + shared.args.extensions = shared.args.extensions or [] + if extension not in shared.args.extensions: + shared.args.extensions.append(extension) +if shared.args.extensions is not None and len(shared.args.extensions) > 0: extensions_module.load_extensions() # Choosing the default model