Loading modules/images.py +1 −0 Original line number Diff line number Diff line Loading @@ -43,6 +43,7 @@ def image_grid(imgs, batch_size=1, rows=None): grid = Image.new('RGB', size=(cols * w, rows * h), color='black') for i, img in enumerate(imgs): script_callbacks.image_grid_loop_callback(img) grid.paste(img, box=(i % cols * w, i // cols * h)) return grid Loading modules/script_callbacks.py +20 −0 Original line number Diff line number Diff line Loading @@ -51,6 +51,11 @@ class UiTrainTabParams: self.txt2img_preview_params = txt2img_preview_params class ImageGridLoopParams: def __init__(self, img): self.img = img ScriptCallback = namedtuple("ScriptCallback", ["script", "callback"]) callback_map = dict( callbacks_app_started=[], Loading @@ -63,6 +68,7 @@ callback_map = dict( callbacks_cfg_denoiser=[], callbacks_before_component=[], callbacks_after_component=[], callbacks_image_grid_loop=[], ) Loading Loading @@ -154,6 +160,12 @@ def after_component_callback(component, **kwargs): except Exception: report_exception(c, 'after_component_callback') def image_grid_loop_callback(component, **kwargs): for c in callback_map['callbacks_image_grid_loop']: try: c.callback(component, **kwargs) except Exception: report_exception(c, 'image_grid_loop') def add_callback(callbacks, fun): stack = [x for x in inspect.stack() if x.filename != __file__] Loading Loading @@ -255,3 +267,11 @@ def on_before_component(callback): def on_after_component(callback): """register a function to be called after a component is created. See on_before_component for more.""" add_callback(callback_map['callbacks_after_component'], callback) def on_image_grid_loop(callback): """register a function to be called inside the image grid loop. The callback is called with one argument: - params: ImageGridLoopParams - parameters to be used inside the image grid loop. """ add_callback(callback_map['callbacks_image_grid_loop'], callback) Loading
modules/images.py +1 −0 Original line number Diff line number Diff line Loading @@ -43,6 +43,7 @@ def image_grid(imgs, batch_size=1, rows=None): grid = Image.new('RGB', size=(cols * w, rows * h), color='black') for i, img in enumerate(imgs): script_callbacks.image_grid_loop_callback(img) grid.paste(img, box=(i % cols * w, i // cols * h)) return grid Loading
modules/script_callbacks.py +20 −0 Original line number Diff line number Diff line Loading @@ -51,6 +51,11 @@ class UiTrainTabParams: self.txt2img_preview_params = txt2img_preview_params class ImageGridLoopParams: def __init__(self, img): self.img = img ScriptCallback = namedtuple("ScriptCallback", ["script", "callback"]) callback_map = dict( callbacks_app_started=[], Loading @@ -63,6 +68,7 @@ callback_map = dict( callbacks_cfg_denoiser=[], callbacks_before_component=[], callbacks_after_component=[], callbacks_image_grid_loop=[], ) Loading Loading @@ -154,6 +160,12 @@ def after_component_callback(component, **kwargs): except Exception: report_exception(c, 'after_component_callback') def image_grid_loop_callback(component, **kwargs): for c in callback_map['callbacks_image_grid_loop']: try: c.callback(component, **kwargs) except Exception: report_exception(c, 'image_grid_loop') def add_callback(callbacks, fun): stack = [x for x in inspect.stack() if x.filename != __file__] Loading Loading @@ -255,3 +267,11 @@ def on_before_component(callback): def on_after_component(callback): """register a function to be called after a component is created. See on_before_component for more.""" add_callback(callback_map['callbacks_after_component'], callback) def on_image_grid_loop(callback): """register a function to be called inside the image grid loop. The callback is called with one argument: - params: ImageGridLoopParams - parameters to be used inside the image grid loop. """ add_callback(callback_map['callbacks_image_grid_loop'], callback)
