diff --git a/extensions-builtin/forge_legacy_preprocessors/legacy_preprocessors/preprocessor_compiled.py b/extensions-builtin/forge_legacy_preprocessors/legacy_preprocessors/preprocessor_compiled.py index e4afbc4e..dc082343 100644 --- a/extensions-builtin/forge_legacy_preprocessors/legacy_preprocessors/preprocessor_compiled.py +++ b/extensions-builtin/forge_legacy_preprocessors/legacy_preprocessors/preprocessor_compiled.py @@ -1233,76 +1233,76 @@ legacy_preprocessors = { "priority": 0, "tags": [] }, - "tile_colorfix": { - "label": "tile_colorfix", - "call_function": identity, - "unload_function": None, - "managed_model": None, - "model_free": False, - "no_control_mode": False, - "resolution": None, - "slider_1": { - "label": "Variation", - "value": 8.0, - "minimum": 3.0, - "maximum": 32.0, - "step": 1.0 - }, - "slider_2": None, - "slider_3": None, - "priority": 0, - "tags": [ - "Tile", "Blur", - ] - }, - "tile_colorfix+sharp": { - "label": "tile_colorfix+sharp", - "call_function": identity, - "unload_function": None, - "managed_model": None, - "model_free": False, - "no_control_mode": False, - "resolution": None, - "slider_1": { - "label": "Variation", - "value": 8.0, - "minimum": 3.0, - "maximum": 32.0, - "step": 1.0 - }, - "slider_2": { - "label": "Sharpness", - "value": 1.0, - "minimum": 0.0, - "maximum": 2.0, - "step": 0.01 - }, - "slider_3": None, - "priority": 0, - "tags": [ - "Tile", "Blur", - ] - }, - "tile_resample": { - "label": "tile_resample", - "call_function": tile_resample, - "unload_function": None, - "managed_model": None, - "model_free": False, - "no_control_mode": False, - "resolution": None, - "slider_1": { - "label": "Down Sampling Rate", - "value": 1.0, - "minimum": 1.0, - "maximum": 8.0, - "step": 0.01 - }, - "slider_2": None, - "slider_3": None, - "priority": 100, - "tags": [ - "Tile", "Blur", - ] - } + # "tile_colorfix": { + # "label": "tile_colorfix", + # "call_function": identity, + # "unload_function": None, + # "managed_model": None, + # "model_free": False, + # "no_control_mode": False, + # "resolution": None, + # "slider_1": { + # "label": "Variation", + # "value": 8.0, + # "minimum": 3.0, + # "maximum": 32.0, + # "step": 1.0 + # }, + # "slider_2": None, + # "slider_3": None, + # "priority": 0, + # "tags": [ + # "Tile", "Blur", + # ] + # }, + # "tile_colorfix+sharp": { + # "label": "tile_colorfix+sharp", + # "call_function": identity, + # "unload_function": None, + # "managed_model": None, + # "model_free": False, + # "no_control_mode": False, + # "resolution": None, + # "slider_1": { + # "label": "Variation", + # "value": 8.0, + # "minimum": 3.0, + # "maximum": 32.0, + # "step": 1.0 + # }, + # "slider_2": { + # "label": "Sharpness", + # "value": 1.0, + # "minimum": 0.0, + # "maximum": 2.0, + # "step": 0.01 + # }, + # "slider_3": None, + # "priority": 0, + # "tags": [ + # "Tile", "Blur", + # ] + # }, + # "tile_resample": { + # "label": "tile_resample", + # "call_function": tile_resample, + # "unload_function": None, + # "managed_model": None, + # "model_free": False, + # "no_control_mode": False, + # "resolution": None, + # "slider_1": { + # "label": "Down Sampling Rate", + # "value": 1.0, + # "minimum": 1.0, + # "maximum": 8.0, + # "step": 0.01 + # }, + # "slider_2": None, + # "slider_3": None, + # "priority": 100, + # "tags": [ + # "Tile", "Blur", + # ] + # } } diff --git a/extensions-builtin/forge_preprocessor_tile/scripts/preprocessor_tile.py b/extensions-builtin/forge_preprocessor_tile/scripts/preprocessor_tile.py new file mode 100644 index 00000000..baf18ee4 --- /dev/null +++ b/extensions-builtin/forge_preprocessor_tile/scripts/preprocessor_tile.py @@ -0,0 +1,35 @@ +import torch + +from modules_forge.supported_preprocessor import Preprocessor, PreprocessorParameter +from modules_forge.forge_util import numpy_to_pytorch, resize_image_with_pad +from modules_forge.shared import add_supported_preprocessor + + +class PreprocessorTile(Preprocessor): + def __init__(self): + super().__init__() + self.name = 'tile_resample' + self.tags = ['Tile'] + self.model_filename_filters = ['tile'] + self.slider_resolution = PreprocessorParameter(visible=False) + + +class PreprocessorTileColorFix(PreprocessorTile): + def __init__(self): + super().__init__() + self.name = 'tile_colorfix' + self.slider_1 = PreprocessorParameter(label='Variation', value=8.0, minimum=3.0, maximum=32.0, step=1.0) + + +class PreprocessorTileColorFixSharp(PreprocessorTileColorFix): + def __init__(self): + super().__init__() + self.name = 'tile_colorfix+sharp' + self.slider_2 = PreprocessorParameter(label='Sharpness', value=1.0, minimum=0.0, maximum=2.0, step=0.01) + + +add_supported_preprocessor(PreprocessorTile()) + +add_supported_preprocessor(PreprocessorTileColorFix()) + +add_supported_preprocessor(PreprocessorTileColorFixSharp())