mirror of
https://github.com/vladmandic/automatic
synced 2026-09-19 09:14:35 +02:00
unified logger
This commit is contained in:
@@ -8,6 +8,7 @@ import os
|
||||
import importlib
|
||||
import gradio as gr
|
||||
from modules import scripts_manager, processing, shared, sd_models, devices
|
||||
from modules import logger
|
||||
|
||||
|
||||
repo = 'https://github.com/vladmandic/IP-Instruct'
|
||||
@@ -57,11 +58,11 @@ class Script(scripts_manager.Script):
|
||||
def run(self, p: processing.StableDiffusionProcessing, query, image, strength, tokens, instruct_guidance, image_guidance): # pylint: disable=arguments-differ
|
||||
supported_model_list = ['sd', 'sdxl', 'sd3']
|
||||
if shared.sd_model_type not in supported_model_list:
|
||||
shared.log.warning(f'IP-Instruct: class={shared.sd_model.__class__.__name__} model={shared.sd_model_type} required={supported_model_list}')
|
||||
logger.log.warning(f'IP-Instruct: class={shared.sd_model.__class__.__name__} model={shared.sd_model_type} required={supported_model_list}')
|
||||
return None
|
||||
self.install()
|
||||
if self.lib is None:
|
||||
shared.log.error('IP-Instruct: failed to import library')
|
||||
logger.log.error('IP-Instruct: failed to import library')
|
||||
return None
|
||||
self.orig_pipe = shared.sd_model
|
||||
if shared.sd_model_type == 'sdxl':
|
||||
@@ -83,8 +84,8 @@ class Script(scripts_manager.Script):
|
||||
ip_ckpt = hf.hf_hub_download(repo_id=repo_id, filename=ckpt, cache_dir=shared.opts.hfcache_dir)
|
||||
ip_model = cls(shared.sd_model, encoder, ip_ckpt, device=devices.device, dtypein=devices.dtype, num_tokens=tokens)
|
||||
processing.fix_seed(p)
|
||||
shared.log.debug(f'IP-Instruct: class={shared.sd_model.__class__.__name__} wrapper={ip_model.__class__.__name__} encoder={encoder} adapter={ckpt}')
|
||||
shared.log.info(f'IP-Instruct: image={image} query="{query}" strength={strength} tokens={tokens} instruct_guidance={instruct_guidance} image_guidance={image_guidance}')
|
||||
logger.log.debug(f'IP-Instruct: class={shared.sd_model.__class__.__name__} wrapper={ip_model.__class__.__name__} encoder={encoder} adapter={ckpt}')
|
||||
logger.log.info(f'IP-Instruct: image={image} query="{query}" strength={strength} tokens={tokens} instruct_guidance={instruct_guidance} image_guidance={image_guidance}')
|
||||
|
||||
image_list = ip_model.generate(
|
||||
query = query,
|
||||
|
||||
Reference in New Issue
Block a user