mirror of
https://github.com/vladmandic/automatic
synced 2026-09-19 09:14:35 +02:00
cleanup logger
This commit is contained in:
@@ -1,6 +1,6 @@
|
||||
import gradio as gr
|
||||
from modules import scripts_manager, processing, shared, sd_models
|
||||
from modules import logger
|
||||
from modules.logger import log
|
||||
|
||||
|
||||
supported_models = ['sdxl']
|
||||
@@ -74,12 +74,12 @@ class Script(scripts_manager.Script):
|
||||
from ligo.segments import segment # pylint: disable=unused-import
|
||||
return True
|
||||
except Exception as e:
|
||||
logger.log.error(f'MoD: {e}')
|
||||
log.error(f'MoD: {e}')
|
||||
return False
|
||||
|
||||
def run(self, p: processing.StableDiffusionProcessing, *args): # pylint: disable=arguments-differ, unused-argument
|
||||
if shared.sd_model_type not in supported_models:
|
||||
logger.log.warning(f'MoD: class={shared.sd_model.__class__.__name__} model={shared.sd_model_type} required={supported_models}')
|
||||
log.warning(f'MoD: class={shared.sd_model.__class__.__name__} model={shared.sd_model_type} required={supported_models}')
|
||||
return None
|
||||
if not self.check_dependencies():
|
||||
return None
|
||||
@@ -109,7 +109,7 @@ class Script(scripts_manager.Script):
|
||||
p.extra_generation_params["MoD Y"] = f'{y_tiles}/{p.task_args["tile_height"]}/{p.task_args["tile_row_overlap"]}'
|
||||
p.keep_prompts = True
|
||||
shared.opts.prompt_attention = 'fixed'
|
||||
logger.log.info(f'MoD: xtiles={x_tiles} ytiles={y_tiles} xoverlap={p.task_args["tile_col_overlap"]} yoverlap={p.task_args["tile_row_overlap"]} xsize={p.task_args["tile_width"]} ysize={p.task_args["tile_height"]}')
|
||||
log.info(f'MoD: xtiles={x_tiles} ytiles={y_tiles} xoverlap={p.task_args["tile_col_overlap"]} yoverlap={p.task_args["tile_row_overlap"]} xsize={p.task_args["tile_width"]} ysize={p.task_args["tile_height"]}')
|
||||
|
||||
shared.sd_model = sd_models.switch_pipe(StableDiffusionXLTilingPipeline, shared.sd_model)
|
||||
sd_models.set_diffuser_options(shared.sd_model)
|
||||
|
||||
Reference in New Issue
Block a user