This commit is contained in:
lllyasviel
2024-01-30 15:41:25 -08:00
parent 66e563fe03
commit 184bb9a042
2 changed files with 107 additions and 72 deletions

View File

@@ -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",
# ]
# }
}

View File

@@ -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())