mirror of
https://github.com/oobabooga/text-generation-webui.git
synced 2024-10-29 21:50:16 +01:00
73 lines
1.8 KiB
Python
73 lines
1.8 KiB
Python
'''
|
|
Copied from: https://github.com/AUTOMATIC1111/stable-diffusion-webui/pull/14184
|
|
'''
|
|
|
|
import inspect
|
|
import warnings
|
|
from functools import wraps
|
|
|
|
import gradio as gr
|
|
|
|
|
|
class GradioDeprecationWarning(DeprecationWarning):
|
|
pass
|
|
|
|
|
|
def repair(grclass):
|
|
if not getattr(grclass, 'EVENTS', None):
|
|
return
|
|
|
|
@wraps(grclass.__init__)
|
|
def __repaired_init__(self, *args, tooltip=None, source=None, original=grclass.__init__, **kwargs):
|
|
if source:
|
|
kwargs["sources"] = [source]
|
|
|
|
allowed_kwargs = inspect.signature(original).parameters
|
|
fixed_kwargs = {}
|
|
for k, v in kwargs.items():
|
|
if k in allowed_kwargs:
|
|
fixed_kwargs[k] = v
|
|
else:
|
|
warnings.warn(f"unexpected argument for {grclass.__name__}: {k}", GradioDeprecationWarning, stacklevel=2)
|
|
|
|
original(self, *args, **fixed_kwargs)
|
|
|
|
self.webui_tooltip = tooltip
|
|
|
|
for event in self.EVENTS:
|
|
replaced_event = getattr(self, str(event))
|
|
|
|
def fun(*xargs, _js=None, replaced_event=replaced_event, **xkwargs):
|
|
if _js:
|
|
xkwargs['js'] = _js
|
|
|
|
return replaced_event(*xargs, **xkwargs)
|
|
|
|
setattr(self, str(event), fun)
|
|
|
|
grclass.__init__ = __repaired_init__
|
|
grclass.update = gr.update
|
|
|
|
|
|
for component in set(gr.components.__all__ + gr.layouts.__all__):
|
|
repair(getattr(gr, component, None))
|
|
|
|
|
|
class Dependency(gr.events.Dependency):
|
|
def __init__(self, *args, **kwargs):
|
|
super().__init__(*args, **kwargs)
|
|
|
|
def then(*xargs, _js=None, **xkwargs):
|
|
if _js:
|
|
xkwargs['js'] = _js
|
|
|
|
return original_then(*xargs, **xkwargs)
|
|
|
|
original_then = self.then
|
|
self.then = then
|
|
|
|
|
|
gr.events.Dependency = Dependency
|
|
|
|
gr.Box = gr.Group
|