Mirrors/text-generation-webui
1
0
Fork 0
mirror of https://github.com/oobabooga/text-generation-webui.git synced 2024-11-26 09:40:20 +01:00
Code Issues Packages Projects Releases Wiki Activity

Update extensions.py

Browse Source
This commit is contained in:
oobabooga 2023-03-24 16:51:27 -03:00 committed by GitHub
parent acdbd6b708
commit 8fad84abc2
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 5 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
modules/extensions.py
View File

@ -1,3 +1,5 @@
import traceback
import gradio as gr import gradio as gr
import extensions import extensions
@ -18,6 +20,7 @@ def load_extensions():
print('Ok.') print('Ok.')
except: except:
print('Fail.') print('Fail.')
traceback.print_exc()
# This iterator returns the extensions in the order specified in the command-line # This iterator returns the extensions in the order specified in the command-line
def iterator(): def iterator():
@ -38,6 +41,7 @@ def apply_extensions(text, typ):
def create_extensions_block(): def create_extensions_block():
global setup_called global setup_called
# Updating the default values # Updating the default values
for extension, name in iterator(): for extension, name in iterator():
if hasattr(extension, 'params'): if hasattr(extension, 'params'):
@ -47,6 +51,7 @@ def create_extensions_block():
extension.params[param] = shared.settings[_id] extension.params[param] = shared.settings[_id]
should_display_ui = False should_display_ui = False
# Running setup function # Running setup function
if not setup_called: if not setup_called:
for extension, name in iterator(): for extension, name in iterator():