mirror of
https://github.com/vladmandic/automatic
synced 2026-09-19 01:04:32 +02:00
unified logger
This commit is contained in:
@@ -3,6 +3,7 @@
|
||||
from huggingface_hub import hf_hub_download
|
||||
from safetensors.torch import load_file
|
||||
from modules import shared, errors, devices
|
||||
from modules import logger
|
||||
from .layerdiffuse_model import TransparentVAEDecoder
|
||||
from .layerdiffuse_loader import load_lora_to_unet, merge_delta_weights_into_unet
|
||||
|
||||
@@ -43,16 +44,16 @@ def apply_layerdiffuse_sdxl_conv(pipeline):
|
||||
def apply_layerdiffuse():
|
||||
try:
|
||||
if shared.sd_model_type == 'sd':
|
||||
shared.log.info(f'LayerDiffuse: class={shared.sd_model.__class__.__name__}')
|
||||
logger.log.info(f'LayerDiffuse: class={shared.sd_model.__class__.__name__}')
|
||||
apply_layerdiffuse_sd15(shared.sd_model)
|
||||
elif shared.sd_model_type == 'sdxl':
|
||||
# shared.log.info(f'LayerDiffuse: class={shared.sd_model.__class__.__name__} type=attn')
|
||||
# logger.log.info(f'LayerDiffuse: class={shared.sd_model.__class__.__name__} type=attn')
|
||||
# apply_layerdiffuse_sdxl_attn(shared.sd_model)
|
||||
shared.log.info(f'LayerDiffuse: class={shared.sd_model.__class__.__name__} type=conv')
|
||||
logger.log.info(f'LayerDiffuse: class={shared.sd_model.__class__.__name__} type=conv')
|
||||
apply_layerdiffuse_sdxl_conv(shared.sd_model)
|
||||
else:
|
||||
shared.log.warning(f'LayerDiffuse: class={shared.sd_model.__class__.__name__} not supported')
|
||||
logger.log.warning(f'LayerDiffuse: class={shared.sd_model.__class__.__name__} not supported')
|
||||
shared.sd_model.layerdiffusion = True
|
||||
except Exception as e:
|
||||
shared.log.error(f'LayerDiffuse: {e}')
|
||||
logger.log.error(f'LayerDiffuse: {e}')
|
||||
errors.display(e, 'LayerDiffuse')
|
||||
|
||||
Reference in New Issue
Block a user