update logging

This commit is contained in:
Vladimir Mandic
2024-09-22 11:17:06 -04:00
parent 2ccf2a0229
commit b9e230c17f
15 changed files with 89 additions and 86 deletions
+1 -1
View File
@@ -74,7 +74,7 @@ class Script(scripts.Script):
shared.sd_model.sd_model_hash = ''
shared.sd_model.sd_model_checkpoint = model
except Exception as e:
shared.log.error(f'Loading CogVideoX: {e}')
shared.log.error(f'Load CogVideoX: {e}')
if debug:
errors.display(e, 'CogVideoX')
if shared.sd_model_type == 'cogvideox' and model != 'None':
+1 -1
View File
@@ -93,7 +93,7 @@ class FaceRestorerYolo(FaceRestoration):
if self.model is None:
model_file = modelloader.load_file_from_url(url=self.model_url, model_dir=self.model_dir, file_name=self.model_name)
if model_file is not None:
shared.log.info(f'Loading: type=FaceHires model={model_file}')
shared.log.info(f'Load: type=FaceHires model={model_file}')
from ultralytics import YOLO # pylint: disable=import-outside-toplevel
self.model = YOLO(model_file)