mirror of
https://github.com/vladmandic/automatic
synced 2026-09-20 01:31:13 +02:00
unified logger
This commit is contained in:
@@ -8,6 +8,7 @@ TODO text2video items:
|
||||
|
||||
import gradio as gr
|
||||
from modules import scripts_manager, processing, shared, images, sd_models, modelloader
|
||||
from modules import logger
|
||||
|
||||
|
||||
MODELS = [
|
||||
@@ -57,20 +58,20 @@ class Script(scripts_manager.Script):
|
||||
if model_name == 'None':
|
||||
return None
|
||||
model = [m for m in MODELS if m['name'] == model_name][0]
|
||||
shared.log.debug(f'Text2Video: model={model} defaults={use_default} frames={num_frames}, video={video_type} duration={duration} loop={gif_loop} pad={mp4_pad} interpolate={mp4_interpolate}')
|
||||
logger.log.debug(f'Text2Video: model={model} defaults={use_default} frames={num_frames}, video={video_type} duration={duration} loop={gif_loop} pad={mp4_pad} interpolate={mp4_interpolate}')
|
||||
|
||||
if model['path'] in shared.opts.sd_model_checkpoint:
|
||||
shared.log.debug(f'Text2Video cached: model={shared.opts.sd_model_checkpoint}')
|
||||
logger.log.debug(f'Text2Video cached: model={shared.opts.sd_model_checkpoint}')
|
||||
else:
|
||||
checkpoint = sd_models.get_closest_checkpoint_match(model['path'])
|
||||
if checkpoint is None:
|
||||
shared.log.debug(f'Text2Video downloading: model={model["path"]}')
|
||||
logger.log.debug(f'Text2Video downloading: model={model["path"]}')
|
||||
checkpoint = modelloader.download_diffusers_model(hub_id=model['path'])
|
||||
sd_models.list_models()
|
||||
if checkpoint is None:
|
||||
shared.log.error(f'Text2Video: failed to find model={model["path"]}')
|
||||
logger.log.error(f'Text2Video: failed to find model={model["path"]}')
|
||||
return None
|
||||
shared.log.debug(f'Text2Video loading: model={checkpoint}')
|
||||
logger.log.debug(f'Text2Video loading: model={checkpoint}')
|
||||
shared.opts.sd_model_checkpoint = checkpoint.name
|
||||
sd_models.reload_model_weights(op='model')
|
||||
|
||||
@@ -83,11 +84,11 @@ class Script(scripts_manager.Script):
|
||||
elif num_frames > 0:
|
||||
p.task_args['num_frames'] = num_frames
|
||||
else:
|
||||
shared.log.error('Text2Video: invalid number of frames')
|
||||
logger.log.error('Text2Video: invalid number of frames')
|
||||
return None
|
||||
|
||||
shared.sd_model = sd_models.set_diffuser_pipe(shared.sd_model, sd_models.DiffusersTaskType.TEXT_2_IMAGE)
|
||||
shared.log.debug(f'Text2Video: args={p.task_args}')
|
||||
logger.log.debug(f'Text2Video: args={p.task_args}')
|
||||
processed = processing.process_images(p)
|
||||
|
||||
if video_type != 'None':
|
||||
|
||||
Reference in New Issue
Block a user