mirror of
https://github.com/vladmandic/automatic
synced 2026-09-18 16:54:33 +02:00
unified logger
This commit is contained in:
@@ -1,5 +1,6 @@
|
||||
import gradio as gr
|
||||
from modules import shared, scripts_manager, sd_models
|
||||
from modules import logger
|
||||
|
||||
|
||||
class Script(scripts_manager.Script):
|
||||
@@ -13,13 +14,13 @@ class Script(scripts_manager.Script):
|
||||
def apply(self):
|
||||
from scripts import layerdiffuse # pylint: disable=no-name-in-module
|
||||
if not shared.sd_loaded:
|
||||
shared.log.error('LayerDiffuse: model not loaded')
|
||||
logger.log.error('LayerDiffuse: model not loaded')
|
||||
return self.is_active()
|
||||
if shared.sd_model_type != 'sd' and shared.sd_model_type != 'sdxl':
|
||||
shared.log.error(f'LayerDiffuse: incorrect base model: class={shared.sd_model.__class__.__name__} type={shared.sd_model_type}')
|
||||
logger.log.error(f'LayerDiffuse: incorrect base model: class={shared.sd_model.__class__.__name__} type={shared.sd_model_type}')
|
||||
return self.is_active()
|
||||
if hasattr(shared.sd_model, 'layerdiffusion'):
|
||||
shared.log.warning('LayerDiffuse: already applied')
|
||||
logger.log.warning('LayerDiffuse: already applied')
|
||||
return self.is_active()
|
||||
layerdiffuse.apply_layerdiffuse()
|
||||
return self.is_active()
|
||||
|
||||
Reference in New Issue
Block a user