Browse Source

Load extensions at startup

oobabooga 3 years ago
parent
commit
1d1f931757
1 changed files with 4 additions and 2 deletions
  1. 4 2
      server.py

+ 4 - 2
server.py

@@ -231,7 +231,6 @@ def apply_extensions(text, typ):
     for ext in sorted(extension_state, key=lambda x : extension_state[x][1]):
     for ext in sorted(extension_state, key=lambda x : extension_state[x][1]):
         if extension_state[ext][0] == True:
         if extension_state[ext][0] == True:
             ext_string = f"extensions.{ext}.script"
             ext_string = f"extensions.{ext}.script"
-            exec(f"import {ext_string}")
             if typ == "input":
             if typ == "input":
                 text = eval(f"{ext_string}.input_modifier(text)")
                 text = eval(f"{ext_string}.input_modifier(text)")
             else:
             else:
@@ -258,8 +257,11 @@ extension_state = {}
 if args.extensions is not None:
 if args.extensions is not None:
     for i,ext in enumerate(args.extensions.split(',')):
     for i,ext in enumerate(args.extensions.split(',')):
         if ext in available_extensions:
         if ext in available_extensions:
-            print(f'The extension "{ext}" is enabled.')
+            print(f'Loading the extension "{ext}"... ', end='')
+            ext_string = f"extensions.{ext}.script"
+            exec(f"import {ext_string}")
             extension_state[ext] = [True, i]
             extension_state[ext] = [True, i]
+            print(f'Ok.')
 
 
 # Choosing the default model
 # Choosing the default model
 if args.model is not None:
 if args.model is not None: